diff options
Diffstat (limited to 'tensorflow/python/training/input.py')
-rw-r--r-- | tensorflow/python/training/input.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/python/training/input.py b/tensorflow/python/training/input.py index 042e68df76..cadb404a83 100644 --- a/tensorflow/python/training/input.py +++ b/tensorflow/python/training/input.py @@ -64,7 +64,8 @@ def match_filenames_once(pattern, name=None): """ with ops.name_scope(name, "matching_filenames", [pattern]) as name: return variables.Variable(io_ops.matching_files(pattern), trainable=False, - name=name, validate_shape=False) + name=name, validate_shape=False, + collections=[ops.GraphKeys.LOCAL_VARIABLES]) def limit_epochs(tensor, num_epochs=None, name=None): |