mirrors
/
PaddleClas
mirror of https://github.com/PaddlePaddle/PaddleClas.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

update resnet.py

pull/2048/head
cuicheng01 2022-06-13 11:28:48 +00:00 committed by Tingquan Gao
parent 50f651e448
commit 4a247f49dc
1 changed files with 3 additions and 1 deletions
ppcls/arch/backbone/legendary_models

4
ppcls/arch/backbone/legendary_models/resnet.py
View File

@ -308,8 +308,10 @@ class ResNet(TheseusLayer):
)), "lr_mult_list should be in (list, tuple) but got {}".format(
type(self.lr_mult_list))
if len(self.lr_mult_list) != 5:
msg = "lr_mult_list length should be 5 but got {}, default lr_mult_list used".format(len(self.lr_mult_list))
msg = "lr_mult_list length should be 5 but got {}, default lr_mult_list used".format(
len(self.lr_mult_list))
logger.warning(msg)
self.lr_mult_list = [1.0, 1.0, 1.0, 1.0, 1.0]
assert isinstance(self.stride_list, (
list, tuple