迁移的时候多迭代几个epoch吧

littlepai · Dec 5, 2019 · 5010ba3 · 5010ba3
1 parent 6f5ee00
commit 5010ba3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/run_tl.sh b/run_tl.sh
@@ -10,4 +10,4 @@ echo "解压一毛一样的迁移样本当验证集（数据少）,存放于 $tl
 unzip data/迁移学习样本.zip -d $tl_val_sample_dir > /dev/null
 
 echo "开始迁移学习"
-python main.py  --restore True --lr 1e-5 --traindir $tl_train_sample_dir --valdir $tl_val_sample_dir --epoch 10 --batchSize 32
+python main.py  --restore True --lr 1e-5 --traindir $tl_train_sample_dir --valdir $tl_val_sample_dir --batchSize 32