From 4a247f49dcc1d3982d08f0ad0a4cdfd517336819 Mon Sep 17 00:00:00 2001 From: cuicheng01 Date: Mon, 13 Jun 2022 11:28:48 +0000 Subject: [PATCH] update resnet.py --- ppcls/arch/backbone/legendary_models/resnet.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/ppcls/arch/backbone/legendary_models/resnet.py b/ppcls/arch/backbone/legendary_models/resnet.py index 48e52ca48..705511f5b 100644 --- a/ppcls/arch/backbone/legendary_models/resnet.py +++ b/ppcls/arch/backbone/legendary_models/resnet.py @@ -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