summaryrefslogtreecommitdiff
path: root/run-flow.sh
diff options
context:
space:
mode:
Diffstat (limited to 'run-flow.sh')
-rwxr-xr-xrun-flow.sh7
1 files changed, 3 insertions, 4 deletions
diff --git a/run-flow.sh b/run-flow.sh
index 858be31..3f650dc 100755
--- a/run-flow.sh
+++ b/run-flow.sh
@@ -6,7 +6,7 @@ dataset="woodflowfixed"
# --fold_A "$HOME/Desktop/thumbs/$dataset/A" \
# --fold_B "$HOME/Desktop/thumbs/$dataset/B" \
# --fold_AB "./datasets/$dataset/"
-#
+
python train.py \
--dataroot "./datasets/$dataset" \
--name "$dataset" \
@@ -17,11 +17,10 @@ python train.py \
--which_direction AtoB \
--lambda_B 100 \
--dataset_mode aligned \
+ --continue_train \
+ --which_epoch latest \
--no_lsgan --norm batch --pool_size 0
-# --continue_train \
-# --which_epoch latest \
-
#python canny-cv.py \
# --dataroot /home/ubuntu/Desktop/thumbs/wood/B/train/ \
# --name woodcanny \