diff --git a/pl_bolts/models/self_supervised/cpc/cpc_module.py b/pl_bolts/models/self_supervised/cpc/cpc_module.py index 926a4f6b25..d325fc55cf 100644 --- 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)