MycroftAI
/
mimic2
mirror of https://github.com/MycroftAI/mimic2.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Merge pull request #61 from greenkeytech/tf1.4fix 

fix for tensorflow 1.4.0
pull/2/head
Keith Ito 2017-10-14 09:28:03 -07:00 committed by GitHub
parent 268995c08c 7b09826434
commit 6612fd3ef6
1 changed files with 16 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
models/helpers.py
View File

@ -15,6 +15,14 @@ class TacoTestHelper(Helper):
def batch_size(self):
return self._batch_size
@property
def sample_ids_shape(self):
return tf.TensorShape([])
@property
def sample_ids_dtype(self):
return np.int32
def initialize(self, name=None):
return (tf.tile([False], [self._batch_size]), _go_frames(self._batch_size, self._output_dim))
@ -48,6 +56,14 @@ class TacoTrainingHelper(Helper):
def batch_size(self):
return self._batch_size
@property
def sample_ids_shape(self):
return tf.TensorShape([])
@property
def sample_ids_dtype(self):
return np.int32
def initialize(self, name=None):
return (tf.tile([False], [self._batch_size]), _go_frames(self._batch_size, self._output_dim))