From 365fee78060021aa39b5d683c2c7d3cfe9e57cd8 Mon Sep 17 00:00:00 2001 From: Bhaskar Gautam Date: Tue, 18 Aug 2020 09:35:31 +0530 Subject: [PATCH] Added support for TF 2.3.0 --- input_pipeline.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/input_pipeline.py b/input_pipeline.py index 5ef8310..036b5f6 100644 --- a/input_pipeline.py +++ b/input_pipeline.py @@ -52,7 +52,7 @@ def input_fn(): # same input file is sent to all workers. if isinstance(input_file, str) or len(input_file) == 1: options = tf.data.Options() - options.experimental_distribute.auto_shard = False + options.experimental_distribute.auto_shard_policy = tf.data.experimental.AutoShardPolicy.OFF d = d.with_options(options) return d