freeze ga

cizhenshi · Jul 26, 2019 · 6e7b47c · 6e7b47c
1 parent 03da01f
commit 6e7b47c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/configs/rscup/htc_deform_ga.py b/configs/rscup/htc_deform_ga.py
@@ -288,7 +288,7 @@
         test_mode=True))
 # optimizer
 optimizer = dict(type='SGD', lr=6e-4, momentum=0.9, weight_decay=0.0001)
-optimizer_config = dict(grad_clip=dict(max_norm=35, norm_type=2))
+optimizer_config = dict(grad_clip=dict(max_norm=35, norm_type=2), paramwise_options=dict(nofreeze='rpn_head'))
 # learning policy
 lr_config = dict(
     policy='step',