Fix hparams forgotten rename in CPCv2 module

self.hparams.encoder was renamed to self.hparams.encoder_name in a8e3fb5 PyTorchLightning#264 Rename overlooked occurrence as in 2163626 PyTorchLightning#680
praecipue · Nov 18, 2021 · 62fbdc9 · 62fbdc9
1 parent b20bba8
commit 62fbdc9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pl_bolts/models/self_supervised/cpc/cpc_module.py b/pl_bolts/models/self_supervised/cpc/cpc_module.py
@@ -124,7 +124,7 @@ def forward(self, img_1):
         Z = self.encoder(img_1)
 
         # non cpc resnets return a list
-        if self.hparams.encoder != "cpc_encoder":
+        if self.hparams.encoder_name != "cpc_encoder":
             Z = Z[0]
 
         # (?) -> (b, -1, nb_feats, nb_feats)