This commit is contained in:
lee
2025-06-19 17:36:24 +08:00
parent bf9604ec29
commit 061820c34f
4 changed files with 15 additions and 11 deletions

View File

@ -205,7 +205,7 @@ class ResNet(nn.Module):
if norm_layer is None:
norm_layer = nn.BatchNorm2d
self._norm_layer = norm_layer
print("ResNet scale: >>>>>>>>>> ", scale)
print("通道剪枝 {}".format(scale))
self.inplanes = 64
self.dilation = 1
if replace_stride_with_dilation is None: