v5 eval

CompVis · Oct 25, 2017 · 57103d1 · 57103d1
1 parent b9f717c
commit 57103d1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/main.py b/main.py
@@ -12,7 +12,7 @@
 
 import nn
 import models
-from batches_pg2 import get_batches, plot_batch, postprocess
+from batches_pg2_vis import get_batches, plot_batch, postprocess
 import deeploss
 
 N_BOXES = 8
@@ -575,7 +575,7 @@ def process_batches(batches):
 
     elif opt.mode == "transfer":
         if not opt.checkpoint:
-            opt.checkpoint = "log/2017-10-24T16:34:09/checkpoints/model.ckpt-100000"
+            opt.checkpoint = "log/2017-10-25T16:31:50/checkpoints/model.ckpt-22000"
         batch_size = opt.batch_size
         img_shape = 2*[opt.spatial_size] + [3]
         data_shape = [batch_size] + img_shape