Merge pull request WZMIAOMIAO#140 from WZMIAOMIAO/dev

update default para
1138886114 · Feb 1, 2021 · 4ef38f3 · 4ef38f3
2 parents 0071b69 + 59f1de0
commit 4ef38f3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch_classification/Test5_resnet/model.py b/pytorch_classification/Test5_resnet/model.py
@@ -87,7 +87,7 @@ def __init__(self,
                  num_classes=1000,
                  include_top=True,
                  groups=1,
-                 width_per_group=4):
+                 width_per_group=64):
         super(ResNet, self).__init__()
         self.include_top = include_top
         self.in_channel = 64