fix "no training data provided" (#69)

* Update 6_train_Quick96.sh Fix "no training data provided" * Update 6_train_SAEHD.sh fix "no training data provided"
2023-01-24 13:27:17 +03:00
parent 05482b3825
commit 8cda581d2c
2 changed files with 3 additions and 3 deletions
@@ -4,7 +4,7 @@ source env.sh
 $DFL_PYTHON "$DFL_SRC/main.py" train \
    --training-data-src-dir "$DFL_WORKSPACE/data_src/aligned" \
    --training-data-dst-dir "$DFL_WORKSPACE/data_dst/aligned" \
-    --pretraining-data-dir "$DFL_ROOT/pretrain_CelebA" \
+    --pretraining-data-dir "$DFL_SRC/pretrain_CelebA" \
    --model-dir "$DFL_WORKSPACE/model" \
    --model SAEHD