Update pl_bolts/models/self_supervised/resnets.py

Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
Lightning-Universe · Nov 16, 2020 · 47e8fdf · 47e8fdf
1 parent 3e2b8f3
commit 47e8fdf
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/pl_bolts/models/self_supervised/resnets.py b/pl_bolts/models/self_supervised/resnets.py
@@ -135,15 +135,15 @@ def __init__(
         self,
         block,
         layers,
-        num_classes=1000,
-        zero_init_residual=False,
-        groups=1,
-        width_per_group=64,
+        num_classes: int = 1000,
+        zero_init_residual: bool = False,
+        groups: int = 1,
+        width_per_group: int = 64,
         replace_stride_with_dilation=None,
         norm_layer=None,
-        return_all_feature_maps=False,
-        first_conv=True,
-        maxpool1=True,
+        return_all_feature_maps: bool = False,
+        first_conv: bool = True,
+        maxpool1: bool = True,
     ):
         super(ResNet, self).__init__()
         if norm_layer is None: