Fix hparams forgotten rename

self.hparams.encoder was renamed to self.hparams.encoder_name in a8e3fb5 Lightning-Universe#264 Rename overlooked occurrence as in 2163626 Lightning-Universe#680
praecipue · Nov 18, 2021 · 717f1c9 · 717f1c9
1 parent 533c09b
commit 717f1c9
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)