提交 a6a76f3d 编写于 作者: G gaotingquan 提交者: Tingquan Gao

fix: args error

上级 ef0eafd6
...@@ -467,7 +467,7 @@ class Inception_V3(TheseusLayer): ...@@ -467,7 +467,7 @@ class Inception_V3(TheseusLayer):
stages_pattern, stages_pattern,
class_num=1000, class_num=1000,
return_patterns=None, return_patterns=None,
stages_pattern=None): return_stages=None):
super().__init__() super().__init__()
self.inception_a_list = config["inception_a"] self.inception_a_list = config["inception_a"]
...@@ -547,7 +547,7 @@ def InceptionV3(pretrained=False, use_ssld=False, **kwargs): ...@@ -547,7 +547,7 @@ def InceptionV3(pretrained=False, use_ssld=False, **kwargs):
if str, means the path of the pretrained model. if str, means the path of the pretrained model.
use_ssld: bool=False. Whether using distillation pretrained model when pretrained=True. use_ssld: bool=False. Whether using distillation pretrained model when pretrained=True.
Returns: Returns:
model: nn.Layer. Specific `InceptionV3` model model: nn.Layer. Specific `InceptionV3` model
""" """
model = Inception_V3( model = Inception_V3(
NET_CONFIG, NET_CONFIG,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册