gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2025-12-17 11:10:20 +01:00
Code Issues Projects Releases Wiki Activity

bug fix - remove import warning when everything was imported successfully + changed global step api to match TF 1.4

Browse Source
This commit is contained in:
Itai Caspi
2017-11-06 17:28:13 +02:00
parent fd103a7b69
commit b40259c61a
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
architectures/tensorflow_components/architecture.py
View File

@@ -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)

3
coach.py
View File

@@ -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()