diff --git a/train.sh b/train.sh index 768dc0f..7bbf305 100755 --- a/train.sh +++ b/train.sh @@ -11,7 +11,7 @@ CUDA_VISIBLE_DEVICES=$devices python -u train.py --dataset ./data \ --cuda \ --epoch 50 \ --loss_epoch_threshold 50 \ ---sketch_loss_coefficie 1.0 \ +--sketch_loss_coefficient 1.0 \ --beam_size 1 \ --seed 90 \ --save ${save_name} \ @@ -21,4 +21,4 @@ CUDA_VISIBLE_DEVICES=$devices python -u train.py --dataset ./data \ --hidden_size 300 \ --lr_scheduler \ --lr_scheduler_gammar 0.5 \ ---att_vec_size 300 > ${save_name}".log" \ No newline at end of file +--att_vec_size 300 > ${save_name}".log"