Skip to content

Commit a9fb7be

Browse files
committed
fix autotune option
1 parent cc6d8de commit a9fb7be

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

efficientnetv2/datasets.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -549,7 +549,7 @@ def _input_fn(self, batch_size, current_host, num_hosts):
549549
ds = ds.prefetch(1)
550550

551551
options = tf.data.Options()
552-
options.experimental_optimization.autotune = True
552+
options.autotune.enabled = True
553553
return ds.with_options(options)
554554

555555

0 commit comments

Comments
 (0)