ibm-granite
/

granite-geospatial-land-surface-temperature

Model card Files Files and versions Community

Tamaragov commited on 8 days ago

Commit

956397c

•

1 Parent(s): 1c8fb3b

Update config.yaml

Updated config.yaml for Colab

Files changed (1) hide show

config.yaml +1 -1

config.yaml CHANGED Viewed

@@ -35,7 +35,7 @@ out_dtype: float32
 data:
   class_path: GenericNonGeoPixelwiseRegressionDataModule
   init_args:
-    batch_size: 64
     num_workers: 8
     train_transform:
       - class_path: albumentations.HorizontalFlip

 data:
   class_path: GenericNonGeoPixelwiseRegressionDataModule
   init_args:
+    batch_size: 1
     num_workers: 8
     train_transform:
       - class_path: albumentations.HorizontalFlip