mirrors/PaddleClas
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleClas.git synced 2025-06-03 21:55:06 +08:00
Code Issues Projects Releases Wiki Activity

fix (#541)

Browse Source 
* Update res2net.py

* Update res2net_vd.py

* Update resnest.py

* Update distillation_models.py
This commit is contained in:
cuicheng01 2021-01-06 21:45:13 +08:00 committed by GitHub
parent aba9738284
commit d3691054ed
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 3 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ppcls/modeling/architectures/distillation_models.py
View File

@ -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):
student_label = self.student(input) student_label = self.student(input)
return teacher_label, student_label return teacher_label, student_label

5
ppcls/modeling/architectures/res2net.py
View File

@ -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

5
ppcls/modeling/architectures/res2net_vd.py
View File

@ -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

2
ppcls/modeling/architectures/resnest.py
View File

@ -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")