diff --git a/pytorch/train.py b/pytorch/train.py
index c0ac2a8c..15f63f1d 100644
--- 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)