bug fix - remove import warning when everything was imported successfully + changed global step api to match TF 1.4
This commit is contained in:
1 parent
fd103a7b69
commit
b40259c61a
2 files changed
+3
-2
No files matched your search
@@ -49,7 +49,7 @@ class TensorFlowArchitecture(Architecture):
|
||||
if self.tp.seed is not None:
|
||||
tf.set_random_seed(self.tp.seed)
|
||||
with tf.variable_scope(self.name, initializer=tf.contrib.layers.xavier_initializer()):
|
||||
self.global_step = tf.contrib.framework.get_or_create_global_step()
|
||||
self.global_step = tf.train.get_or_create_global_step()
|
||||
|
||||
# build the network
|
||||
self.get_model(tuning_parameters)
|
||||
|
||||
@@ -30,7 +30,8 @@ from subprocess import Popen
|
||||
import datetime
|
||||
import presets
|
||||
|
||||
screen.warning("Warning: failed to import the following packages - {}".format(', '.join(set(failed_imports))))
|
||||
if len(set(failed_imports)) > 0:
|
||||
screen.warning("Warning: failed to import the following packages - {}".format(', '.join(set(failed_imports))))
|
||||
|
||||
time_started = datetime.datetime.now()
|
||||
cur_time = time_started.time()
|
||||
|
||||
Reference in new issue
Block a user