From a6a76f3d3620107e28a921471bebb52074599457 Mon Sep 17 00:00:00 2001 From: gaotingquan Date: Mon, 10 Jan 2022 02:39:18 +0000 Subject: [PATCH] fix: args error --- ppcls/arch/backbone/legendary_models/inception_v3.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ppcls/arch/backbone/legendary_models/inception_v3.py b/ppcls/arch/backbone/legendary_models/inception_v3.py index 6901ba64..5575f8c9 100644 --- a/ppcls/arch/backbone/legendary_models/inception_v3.py +++ b/ppcls/arch/backbone/legendary_models/inception_v3.py @@ -467,7 +467,7 @@ class Inception_V3(TheseusLayer): stages_pattern, class_num=1000, return_patterns=None, - stages_pattern=None): + return_stages=None): super().__init__() self.inception_a_list = config["inception_a"] @@ -547,7 +547,7 @@ def InceptionV3(pretrained=False, use_ssld=False, **kwargs): if str, means the path of the pretrained model. use_ssld: bool=False. Whether using distillation pretrained model when pretrained=True. Returns: - model: nn.Layer. Specific `InceptionV3` model + model: nn.Layer. Specific `InceptionV3` model """ model = Inception_V3( NET_CONFIG, -- GitLab