From 8b3e2316e5dfb31343d47ad8ba10775cbdc0932c Mon Sep 17 00:00:00 2001 From: "yunji.cjy" Date: Tue, 7 Jun 2022 20:38:42 +0800 Subject: [PATCH 1/2] bugfix_timm_init_weights --- easycv/models/backbones/pytorch_image_models_wrapper.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/easycv/models/backbones/pytorch_image_models_wrapper.py b/easycv/models/backbones/pytorch_image_models_wrapper.py index 2bba3ef7..848cf8e4 100644 --- a/easycv/models/backbones/pytorch_image_models_wrapper.py +++ b/easycv/models/backbones/pytorch_image_models_wrapper.py @@ -164,9 +164,6 @@ def init_weights(self, pretrained=None): raise ValueError( 'Error: Fail to create {} with (pretrained={}...)'.format( self.model_name, pretrained)) - else: - print_log('load model from init weights') - self.model.init_weights() def forward(self, x): From 14b17b2e9c0de4abbbf03aee733ac9768e347502 Mon Sep 17 00:00:00 2001 From: "yunji.cjy" Date: Thu, 9 Jun 2022 16:22:16 +0800 Subject: [PATCH 2/2] update --- easycv/models/backbones/pytorch_image_models_wrapper.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/easycv/models/backbones/pytorch_image_models_wrapper.py b/easycv/models/backbones/pytorch_image_models_wrapper.py index 848cf8e4..e7fa8b85 100644 --- a/easycv/models/backbones/pytorch_image_models_wrapper.py +++ b/easycv/models/backbones/pytorch_image_models_wrapper.py @@ -164,6 +164,8 @@ def init_weights(self, pretrained=None): raise ValueError( 'Error: Fail to create {} with (pretrained={}...)'.format( self.model_name, pretrained)) + else: + print_log('load model from init weights') def forward(self, x):