diff --git a/conf/vhr10.yaml b/conf/vhr10.yaml index c3eac82fbb0..d4bdf2ac1c7 100644 --- 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 index e7262e6ce25..f500d21b9e4 100644 --- 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