Add try_gcs flag. Fix #1054

mingxingtan · mingxingtan · commit 0fb012a80487 · 2021-07-16T11:00:48.000-07:00
diff --git a/efficientnetv2/datasets.py b/efficientnetv2/datasets.py
@@ -508,6 +508,7 @@ class CIFAR10Input(ImageNetInput):
   cfg = copy.deepcopy(ImageNetInput.cfg)
   cfg.update(
       dict(
+          try_gcs=True,
           data_dir=None,
           num_classes=10,
           multiclass=False,
