...

LynnHo · Apr 11, 2018 · c3a4e56 · c3a4e56
1 parent e009f7e
commit c3a4e56
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch/train.py b/pytorch/train.py
@@ -67,7 +67,7 @@
 
 
 """ load checkpoint """
-ckpt_dir = './checkpoints/celeba_dcgan'
+ckpt_dir = './checkpoints/horse2zebra'
 utils.mkdir(ckpt_dir)
 try:
     ckpt = utils.load_checkpoint(ckpt_dir)