mirror of
https://github.com/twitter/the-algorithm.git
synced 2025-02-13 02:48:56 +01:00
Merge 88240565276f606ec7ef43419ebe58ab4cdbe55d into 72eda9a24f815f6d566818cbf8518138e29d83e9
This commit is contained in:
commit
ef7e418753
@ -118,7 +118,7 @@ def data_record_input_fn(
|
||||
|
||||
if initializable:
|
||||
# when the data parsing dpends on some HashTable or Tensor, the iterator is initalizable and
|
||||
# therefore we need to be run explicitly
|
||||
# therefore it needs to be run explicitly
|
||||
iterator = dataset.make_initializable_iterator()
|
||||
tf.add_to_collection(tf.GraphKeys.TABLE_INITIALIZERS, iterator.initializer)
|
||||
else:
|
||||
|
Loading…
x
Reference in New Issue
Block a user