added a dataset, tried testing (unsuccessfully)

1 files changed, 2 insertions, 1 deletions

diff --git a/main.py b/main.py
index 0d84f216..063670b8 100644
--- a/main.py
+++ b/main.py
@@ -42,7 +42,7 @@ def parse_args():
     return parser.parse_args()
 
 def train(model):
-    for _ in hp.num_epochs:
+    for _ in range(hp.num_epochs):
         model.train_step()
 
 def main():
@@ -52,6 +52,7 @@ def main():
     if os.path.exists(ARGS.style):
         ARGS.style = os.path.abspath(ARGS.style)
     os.chdir(sys.path[0])
+    print('this is',ARGS.content)
 
     content_image = imread(ARGS.content)
     style_image = imread(ARGS.style)