"git@developer.sourcefind.cn:orangecat/ollama.git" did not exist on "883ec4d1efc0a82f7fee1699a3eca8a046184b4c"
Commit 7acce35d authored by Jing Li's avatar Jing Li Committed by A. Unique TensorFlower
Browse files

Fix bug in test dataset generation

PiperOrigin-RevId: 273066504
parent f8d9c9b8
...@@ -130,7 +130,8 @@ def get_input_dataset(flags_obj, strategy): ...@@ -130,7 +130,8 @@ def get_input_dataset(flags_obj, strategy):
if strategy: if strategy:
if isinstance(strategy, tf.distribute.experimental.TPUStrategy): if isinstance(strategy, tf.distribute.experimental.TPUStrategy):
test_ds = strategy.experimental_distribute_datasets_from_function(_test_data_fn) test_ds = strategy.experimental_distribute_datasets_from_function(
_test_data_fn)
else: else:
test_ds = strategy.experimental_distribute_dataset(_test_data_fn()) test_ds = strategy.experimental_distribute_dataset(_test_data_fn())
else: else:
......
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