Skip to content

GitLab

Commit 868cea8f authored by vishnubanna's avatar vishnubanna
Browse files

fixed distribution item

parent 47545935
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
orbit/utils/common.py
View file @ 868cea8f
...@@ -82,8 +82,7 @@ def make_distributed_dataset(strategy, dataset_or_fn, *args, **kwargs): ...@@ -82,8 +82,7 @@ def make_distributed_dataset(strategy, dataset_or_fn, *args, **kwargs):
kwargs["input_context"] = input_context kwargs["input_context"] = input_context
return dataset_or_fn(*args, **kwargs) return dataset_or_fn(*args, **kwargs)
return strategy.experimental_distribute_datasets_from_function(dataset_fn) return strategy.distribute_datasets_from_function(dataset_fn)
#return strategy.distribute_datasets_from_function(dataset_fn)
def get_value(x): def get_value(x):
......
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