From 57103d124a602bc7a5b22b61aaca2bedc31b37bb Mon Sep 17 00:00:00 2001 From: Patrick Esser Date: Thu, 26 Oct 2017 00:03:46 +0200 Subject: [PATCH] v5 eval --- main.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/main.py b/main.py index 958e683f..173fd6bd 100644 --- 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