From 8cda581d2c28a8e57b3686c4327858e277f52447 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Claudio=20S=C3=A1nchez?= Date: Tue, 24 Jan 2023 10:27:17 +0000 Subject: [PATCH] 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" --- scripts/6_train_Quick96.sh | 4 ++-- scripts/6_train_SAEHD.sh | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/scripts/6_train_Quick96.sh b/scripts/6_train_Quick96.sh index c958f1b..507ef89 100755 --- a/scripts/6_train_Quick96.sh +++ b/scripts/6_train_Quick96.sh @@ -4,8 +4,8 @@ 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" \ - --pretrained-model-dir "$DFL_ROOT/pretrain_Quick96" \ + --pretraining-data-dir "$DFL_SRC/pretrain_CelebA" \ + --pretrained-model-dir "$DFL_SRC/pretrain_Quick96" \ --model-dir "$DFL_WORKSPACE/model" \ --model Quick96 diff --git a/scripts/6_train_SAEHD.sh b/scripts/6_train_SAEHD.sh index 10ea707..b82aceb 100755 --- a/scripts/6_train_SAEHD.sh +++ b/scripts/6_train_SAEHD.sh @@ -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