From 78fc5f9c078e18dc312b398bdb020029f1f48dd9 Mon Sep 17 00:00:00 2001 From: Zhiyuan Chen Date: Wed, 24 Mar 2021 17:01:36 +0800 Subject: [PATCH] fixes #283 --- tools/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/train.py b/tools/train.py index 16ee091a..8e86c310 100644 --- a/tools/train.py +++ b/tools/train.py @@ -79,7 +79,7 @@ def build_opt_lr(model, current_epoch=0): m.eval() if current_epoch >= cfg.BACKBONE.TRAIN_EPOCH: for layer in cfg.BACKBONE.TRAIN_LAYERS: - for param in getattr(model.backbone, layer).parameters(): + for param in getattr(model.backbone, layer)().parameters(): param.requires_grad = True for m in getattr(model.backbone, layer).modules(): if isinstance(m, nn.BatchNorm2d):