diff --git a/pl_bolts/models/self_supervised/cpc/cpc_module.py b/pl_bolts/models/self_supervised/cpc/cpc_module.py index 71e6bec967..0d7563b045 100644 --- a/pl_bolts/models/self_supervised/cpc/cpc_module.py +++ b/pl_bolts/models/self_supervised/cpc/cpc_module.py @@ -100,7 +100,7 @@ def __compute_final_nb_c(self, patch_size): dummy_batch = self.encoder(dummy_batch) # other encoders return a list - if self.hparams.encoder != 'cpc_encoder': + if self.hparams.encoder_name != 'cpc_encoder': dummy_batch = dummy_batch[0] dummy_batch = self.__recover_z_shape(dummy_batch, 2)