romeokienzler Tamaragov commited on
Commit
475b271
1 Parent(s): 1c8fb3b

Update config.yaml (#20)

Browse files

- Update config.yaml (956397cdd8c3f99f682638729bade51517f34df5)


Co-authored-by: Tamara Govindasamy <[email protected]>

Files changed (1) hide show
  1. config.yaml +1 -1
config.yaml CHANGED
@@ -35,7 +35,7 @@ out_dtype: float32
35
  data:
36
  class_path: GenericNonGeoPixelwiseRegressionDataModule
37
  init_args:
38
- batch_size: 64
39
  num_workers: 8
40
  train_transform:
41
  - class_path: albumentations.HorizontalFlip
 
35
  data:
36
  class_path: GenericNonGeoPixelwiseRegressionDataModule
37
  init_args:
38
+ batch_size: 1
39
  num_workers: 8
40
  train_transform:
41
  - class_path: albumentations.HorizontalFlip