gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2025-12-17 19:20:19 +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: if self.tp.seed is not None:
tf.set_random_seed(self.tp.seed) tf.set_random_seed(self.tp.seed)
with tf.variable_scope(self.name, initializer=tf.contrib.layers.xavier_initializer()): 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 # build the network
self.get_model(tuning_parameters) self.get_model(tuning_parameters)

1
coach.py
View File

@@ -30,6 +30,7 @@ from subprocess import Popen
import datetime import datetime
import presets import presets
if len(set(failed_imports)) > 0:
screen.warning("Warning: failed to import the following packages - {}".format(', '.join(set(failed_imports)))) screen.warning("Warning: failed to import the following packages - {}".format(', '.join(set(failed_imports))))
time_started = datetime.datetime.now() time_started = datetime.datetime.now()