未验证 提交 4aeb9a48 编写于 作者: C cuicheng01 提交者: GitHub

fix some bugs (#542)

* Update res2net.py

* Update res2net_vd.py

* Update resnest.py

* Update distillation_models.py
上级 fc84d99a
...@@ -50,7 +50,7 @@ class ResNet50_vd_distill_MobileNetV3_large_x1_0(nn.Layer): ...@@ -50,7 +50,7 @@ class ResNet50_vd_distill_MobileNetV3_large_x1_0(nn.Layer):
class ResNeXt101_32x16d_wsl_distill_ResNet50_vd(nn.Layer): class ResNeXt101_32x16d_wsl_distill_ResNet50_vd(nn.Layer):
def __init__(self, class_dim=1000, **args): def __init__(self, class_dim=1000, **args):
super(ResNet50_vd_distill_MobileNetV3_large_x1_0, self).__init__() super(ResNeXt101_32x16d_wsl_distill_ResNet50_vd, self).__init__()
self.teacher = ResNeXt101_32x16d_wsl(class_dim=class_dim, **args) self.teacher = ResNeXt101_32x16d_wsl(class_dim=class_dim, **args)
...@@ -62,4 +62,4 @@ class ResNeXt101_32x16d_wsl_distill_ResNet50_vd(nn.Layer): ...@@ -62,4 +62,4 @@ class ResNeXt101_32x16d_wsl_distill_ResNet50_vd(nn.Layer):
student_label = self.student(input) student_label = self.student(input)
return teacher_label, student_label return teacher_label, student_label
\ No newline at end of file
...@@ -247,11 +247,6 @@ def Res2Net50_14w_8s(**args): ...@@ -247,11 +247,6 @@ def Res2Net50_14w_8s(**args):
return model return model
def Res2Net50_48w_2s(**args):
model = Res2Net(layers=50, scales=2, width=48, **args)
return model
def Res2Net50_26w_6s(**args): def Res2Net50_26w_6s(**args):
model = Res2Net(layers=50, scales=6, width=26, **args) model = Res2Net(layers=50, scales=6, width=26, **args)
return model return model
......
...@@ -270,11 +270,6 @@ def Res2Net50_vd_14w_8s(**args): ...@@ -270,11 +270,6 @@ def Res2Net50_vd_14w_8s(**args):
return model return model
def Res2Net50_vd_48w_2s(**args):
model = Res2Net_vd(layers=50, scales=2, width=48, **args)
return model
def Res2Net50_vd_26w_6s(**args): def Res2Net50_vd_26w_6s(**args):
model = Res2Net_vd(layers=50, scales=6, width=26, **args) model = Res2Net_vd(layers=50, scales=6, width=26, **args)
return model return model
......
...@@ -242,7 +242,7 @@ class BottleneckBlock(nn.Layer): ...@@ -242,7 +242,7 @@ class BottleneckBlock(nn.Layer):
num_filters=group_width, num_filters=group_width,
filter_size=3, filter_size=3,
stride=1, stride=1,
dilation=dialtion, dilation=dilation,
groups=cardinality, groups=cardinality,
act="relu", act="relu",
name=name + "_conv2") name=name + "_conv2")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册