Added keyword when loading from file

bob/learn/tensorflow/trainers/SiameseTrainer.py

@@ -159,10 +159,10 @@ class SiameseTrainer(Trainer):
         # Creating the variables
         tf.global_variables_initializer().run(session=self.session)
 
-    def create_network_from_file(self, model_from_file):
+    def create_network_from_file(self, model_from_file, clear_devices=True):
 
         #saver = self.architecture.load(self.model_from_file, clear_devices=False)
-        self.saver = tf.train.import_meta_graph(model_from_file + ".meta")
+        self.saver = tf.train.import_meta_graph(model_from_file + ".meta", clear_devices=clear_devices)
         self.saver.restore(self.session, model_from_file)
 
         # Loading the graph from the graph pointers

bob/learn/tensorflow/trainers/Trainer.py

@@ -164,7 +164,7 @@ class Trainer(object):
         # Creating the variables
         tf.global_variables_initializer().run(session=self.session)
 
-    def create_network_from_file(self, file_name):
+    def create_network_from_file(self, file_name, clear_devices=True):
         """
         Bootstrap a graph from a checkpoint
 
@@ -172,7 +172,7 @@ class Trainer(object):
 
            file_name: Name of of the checkpoing
         """
-        self.saver = tf.train.import_meta_graph(file_name + ".meta")
+        self.saver = tf.train.import_meta_graph(file_name + ".meta", clear_devices=clear_devices)
         self.saver.restore(self.session, file_name)
 
         # Loading training graph

bob/learn/tensorflow/trainers/TripletTrainer.py

@@ -178,10 +178,10 @@ class TripletTrainer(Trainer):
         # Creating the variables
         tf.global_variables_initializer().run(session=self.session)
 
-    def create_network_from_file(self, model_from_file):
+    def create_network_from_file(self, model_from_file, clear_devices=True):
 
         #saver = self.architecture.load(self.model_from_file, clear_devices=False)
-        self.saver = tf.train.import_meta_graph(model_from_file + ".meta")
+        self.saver = tf.train.import_meta_graph(model_from_file + ".meta", clear_devices=clear_devices)
         self.saver.restore(self.session, model_from_file)
 
         # Loading the graph from the graph pointers