From 3f9b0e206424cdcb27ed3ea4c1396bcb50d53c82 Mon Sep 17 00:00:00 2001 From: Bhavya Date: Fri, 28 Feb 2020 15:06:19 +0530 Subject: [PATCH] Update train.sh --- train.sh | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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"