Update conf

microsoft · Feb 2, 2023 · f27892d · f27892d
1 parent 18a2fa4
commit f27892d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/conf/vhr10.yaml b/conf/vhr10.yaml
@@ -15,10 +15,9 @@ experiment:
   name: "vhr10_test"
   module:
     model: "faster-rcnn"
-    backbone: "resnet18"
-    pretrained: True
+    backbone: "resnet50"
     num_classes: 11
-    learning_rate: 1.3e-5
+    learning_rate: 1.6e-5
     learning_rate_schedule_patience: 6
     verbose: false
   datamodule:

diff --git a/tests/conf/vhr10.yaml b/tests/conf/vhr10.yaml
@@ -1,7 +1,7 @@
 experiment:
   task: "vhr10"
   module:
-    detection_model: "faster-rcnn"
+    model: "faster-rcnn"
     backbone: "resnet18"
     num_classes: 11
     learning_rate: 1e-4