fixed cpc module issue (Lightning-Universe#680)

Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com>
praecipue · Jul 4, 2021 · 2163626 · 2163626
1 parent 270867c
commit 2163626
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
@@ -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)