Merge pull request #26 from ST2-EV/patch-1

replacing fs for load_from_disk with storage_options

scripts/train_unet.py

@@ -53,7 +53,7 @@ def main(args):
     if args.dataset_name is not None:
         if os.path.exists(args.dataset_name):
             dataset = load_from_disk(args.dataset_name,
-                                     args.dataset_config_name)["train"]
+                                     storage_options = args.dataset_config_name)["train"]
         else:
             dataset = load_dataset(
                 args.dataset_name,