최강혁 / dddd

Go to a project
Toggle navigation Toggle navigation pinning
yunjey
70391946 1 parent 053467a8

train and eval the model

Inline Side-by-side
Showing 1 changed file with 1 additions and 0 deletions
...@@ -12,6 +12,7 @@ class Solver(object): ...@@ -12,6 +12,7 @@ class Solver(object):
12 def __init__(self, model, batch_size=100, pretrain_iter=10000, train_iter=2000, sample_iter=100, 12 def __init__(self, model, batch_size=100, pretrain_iter=10000, train_iter=2000, sample_iter=100,
13 svhn_dir='svhn', mnist_dir='mnist', log_dir='logs', sample_save_path='sample', 13 svhn_dir='svhn', mnist_dir='mnist', log_dir='logs', sample_save_path='sample',
14 model_save_path='model', pretrained_model='model/svhn_model-10000', test_model='model/dtn-2000'): 14 model_save_path='model', pretrained_model='model/svhn_model-10000', test_model='model/dtn-2000'):
15 +
15 self.model = model 16 self.model = model
16 self.batch_size = batch_size 17 self.batch_size = batch_size
17 self.pretrain_iter = pretrain_iter 18 self.pretrain_iter = pretrain_iter
......