Skip to content

GitLab

"git@developer.sourcefind.cn:change/sglang.git" did not exist on "4a102a2b02f120fc84c481019e35b2b041ce06f7"
Commit 2c810902 authored by Chen Chen's avatar Chen Chen Committed by A. Unique TensorFlower
Browse files

Internal Change 

PiperOrigin-RevId: 319282071
parent 527ce298
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
official/core/input_reader.py
View file @ 2c810902
...@@ -171,6 +171,9 @@ class InputReader: ...@@ -171,6 +171,9 @@ class InputReader:
as_supervised=self._tfds_as_supervised, as_supervised=self._tfds_as_supervised,
decoders=decoders, decoders=decoders,
read_config=read_config) read_config=read_config)
if self._is_training:
dataset = dataset.repeat()
return dataset return dataset
@property @property
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment