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)