训练脚本纠错

littlepai · Dec 5, 2019 · e84bcb8 · e84bcb8
1 parent 6024f3d
commit e84bcb8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/run.sh b/run.sh
@@ -10,7 +10,7 @@ echo "开始生成训练数据 $train_sample_num 条，存于 $train_sample_di
 python data_gen.py $train_sample_num  $train_sample_dir
 
 echo "开始生成验证数据 $val_sample_num 条，存于 $val_sample_dir"
-python data_gen.py $val_sample_num $val_sample_num
+python data_gen.py $val_sample_num $val_sample_dir
 
 python main.py --epoch 10
 
diff --git a/run_tl.sh b/run_tl.sh
@@ -5,8 +5,8 @@ rm -rf $tl_train_sample_dir
 echo "解压迁移样本当训练集,存放于 $tl_train_sample_dir"
 unzip data/迁移学习样本.zip -d $tl_train_sample_dir > /dev/null
 
-echo "解压一毛一样的迁移样本当验证集（数据少）,存放于 $tl_val_sample_dir"
 rm -rf $tl_val_sample_dir
+echo "解压一毛一样的迁移样本当验证集（数据少）,存放于 $tl_val_sample_dir"
 unzip data/迁移学习样本.zip -d $tl_val_sample_dir > /dev/null
 
 echo "开始迁移学习"