Changed the validation snapshot default

parent 8fd43642
...@@ -60,7 +60,7 @@ class Trainer(object): ...@@ -60,7 +60,7 @@ class Trainer(object):
###### training options ########## ###### training options ##########
iterations=5000, iterations=5000,
snapshot=1000, snapshot=1000,
validation_snapshot=2000, validation_snapshot=2000,#2000,
keep_checkpoint_every_n_hours=2, keep_checkpoint_every_n_hours=2,
## Analizer ## Analizer
...@@ -421,6 +421,7 @@ class Trainer(object): ...@@ -421,6 +421,7 @@ class Trainer(object):
step: Iteration number step: Iteration number
""" """
if self.validation_data_shuffler.prefetch: if self.validation_data_shuffler.prefetch:
embedding, labels = self.session.run([self.validation_predictor, self.validation_label_ph]) embedding, labels = self.session.run([self.validation_predictor, self.validation_label_ph])
else: else:
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment