mirror of
https://github.com/twitter/the-algorithm.git
synced 2025-01-05 00:51:55 +01:00
Compare commits
2 Commits
ef7e418753
...
e1bc65fa68
Author | SHA1 | Date | |
---|---|---|---|
|
e1bc65fa68 | ||
|
8824056527 |
@ -118,7 +118,7 @@ def data_record_input_fn(
|
|||||||
|
|
||||||
if initializable:
|
if initializable:
|
||||||
# when the data parsing dpends on some HashTable or Tensor, the iterator is initalizable and
|
# 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()
|
iterator = dataset.make_initializable_iterator()
|
||||||
tf.add_to_collection(tf.GraphKeys.TABLE_INITIALIZERS, iterator.initializer)
|
tf.add_to_collection(tf.GraphKeys.TABLE_INITIALIZERS, iterator.initializer)
|
||||||
else:
|
else:
|
||||||
|
Loading…
Reference in New Issue
Block a user