Remove auto_lr_find

microsoft · Apr 11, 2023 · a49669c · a49669c
1 parent 4078076
commit a49669c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git 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
@@ -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