diff --git a/conf/nasa_marine_debris.yaml b/conf/nasa_marine_debris.yaml index 4908400bec1..a953056533f 100644 --- a/conf/nasa_marine_debris.yaml +++ b/conf/nasa_marine_debris.yaml @@ -6,7 +6,6 @@ trainer: devices: 1 min_epochs: 5 max_epochs: 100 - auto_lr_find: False benchmark: True experiment: diff --git a/conf/vhr10.yaml b/conf/vhr10.yaml index d4bdf2ac1c7..1da780e19d3 100644 --- a/conf/vhr10.yaml +++ b/conf/vhr10.yaml @@ -7,7 +7,6 @@ trainer: devices: 1 min_epochs: 5 max_epochs: 100 - auto_lr_find: False benchmark: True experiment: @@ -22,6 +21,6 @@ experiment: verbose: false datamodule: root: "data/vhr10" - batch_size: 3 - num_workers: 12 + batch_size: 2 + num_workers: 56 patch_size: 512