mirror of
https://github.com/twitter/the-algorithm.git
synced 2025-01-02 23:51:53 +01:00
Merge 8824056527
into 72eda9a24f
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…
Reference in New Issue
Block a user