step5

Author: yanx27

transfer.py

@@ -14,7 +14,7 @@
 from util import html
 import src.config.test_opt as opt
 
-os.environ['CUDA_VISIBLE_DEVICES'] = "3"
+os.environ['CUDA_VISIBLE_DEVICES'] = "0"
 iter_path = os.path.join(opt.checkpoints_dir, opt.name, 'iter.txt')
 
 data_loader = CreateDataLoader(opt)
@@ -35,4 +35,4 @@
     img_path = data['path']
     visualizer.save_images(webpage, visuals, img_path)
 webpage.save()
-torch.cuda.empty_cache()
+torch.cuda.empty_cache()