jfzhang95 / pytorch-deeplab-xception

DeepLab v3+ model in PyTorch. Support different backbones.
MIT License
2.9k stars 779 forks source link

Question about function "get_1x_lr_params" #197

Open henanjun opened 3 years ago

henanjun commented 3 years ago

Hi

It seems that the function get_1x_lr_params can only Iterate nn.Conv2d in the top node of the backbone. such as the conv1 , while fails to Iterate 'nn.Conv2d' in subnode, such as ('layer3.4', Bottleneck( (conv1): Conv2d(1024, 256, kernel_size=(1, 1), stride=(1, 1), bias=False) (bn1): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True) (conv2): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(2, 2), dilation=(2, 2), bias=False) (bn2): BatchNorm2d(256, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True) (conv3): Conv2d(256, 1024, kernel_size=(1, 1), stride=(1, 1), bias=False) (bn3): BatchNorm2d(1024, eps=1e-05, momentum=0.1, affine=True, track_running_stats=True) (relu): ReLU(inplace=True) ))

so the parameters in the Bottleneck will not update during the training?