mirror of
https://github.com/twitter/the-algorithm.git
synced 2025-01-02 23:51:53 +01:00
Compare commits
2 Commits
ef7e418753
...
e1bc65fa68
Author | SHA1 | Date | |
---|---|---|---|
|
e1bc65fa68 | ||
|
8824056527 |
@ -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