Commit f3dc50ca authored by Tiago de Freitas Pereira's avatar Tiago de Freitas Pereira
Browse files

debugging issue #32

parent ce6435eb
Pipeline #11883 failed with stages
in 13 minutes and 42 seconds
...@@ -165,6 +165,7 @@ class Trainer(object): ...@@ -165,6 +165,7 @@ class Trainer(object):
self.summaries_validation = tf.add_to_collection("summaries_validation", self.summaries_validation) self.summaries_validation = tf.add_to_collection("summaries_validation", self.summaries_validation)
# Creating the variables # Creating the variables
self.session.run(tf.local_variables_initializer())
tf.global_variables_initializer().run(session=self.session) tf.global_variables_initializer().run(session=self.session)
def create_network_from_file(self, file_name, clear_devices=True): def create_network_from_file(self, file_name, clear_devices=True):
...@@ -329,9 +330,8 @@ class Trainer(object): ...@@ -329,9 +330,8 @@ class Trainer(object):
# TODO: JUST FOR TESTING THE INTEGRATION # TODO: JUST FOR TESTING THE INTEGRATION
import ipdb; ipdb.set_trace(); #import ipdb; ipdb.set_trace();
if isinstance(self.train_data_shuffler, TFRecord): if isinstance(self.train_data_shuffler, TFRecord):
self.session.run(tf.local_variables_initializer())
self.thread_pool = tf.train.Coordinator() self.thread_pool = tf.train.Coordinator()
threads = tf.train.start_queue_runners(coord=self.thread_pool, sess=self.session) threads = tf.train.start_queue_runners(coord=self.thread_pool, sess=self.session)
......
Supports Markdown
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