1 files changed, 3 insertions, 4 deletions
diff --git a/run-flow.sh b/run-flow.sh
index 7b7ac4f..3f650dc 100755
--- a/run-flow.sh
+++ b/run-flow.sh
@@ -1,4 +1,4 @@
-dataset="woodflow"
+dataset="woodflowfixed"
 
 #/home/lens/Desktop/dataset-random.pl
 
@@ -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 \