next

cyclegan/test.py

@@ -32,7 +32,7 @@ from data import create_dataset
 from models import create_model
 from util.visualizer import save_images
 from util import html
-
+import torch
 
 if __name__ == '__main__':
     opt = TestOptions().parse()  # get test options