未验证 提交 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):
class ResNeXt101_32x16d_wsl_distill_ResNet50_vd(nn.Layer):
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)
......@@ -62,4 +62,4 @@ class ResNeXt101_32x16d_wsl_distill_ResNet50_vd(nn.Layer):
student_label = self.student(input)
return teacher_label, student_label
\ No newline at end of file
return teacher_label, student_label
......@@ -247,11 +247,6 @@ def Res2Net50_14w_8s(**args):
return model
def Res2Net50_48w_2s(**args):
model = Res2Net(layers=50, scales=2, width=48, **args)
return model
def Res2Net50_26w_6s(**args):
model = Res2Net(layers=50, scales=6, width=26, **args)
return model
......
......@@ -270,11 +270,6 @@ def Res2Net50_vd_14w_8s(**args):
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):
model = Res2Net_vd(layers=50, scales=6, width=26, **args)
return model
......
......@@ -242,7 +242,7 @@ class BottleneckBlock(nn.Layer):
num_filters=group_width,
filter_size=3,
stride=1,
dilation=dialtion,
dilation=dilation,
groups=cardinality,
act="relu",
name=name + "_conv2")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册