diff --git a/pl_bolts/models/rl/dqn_model.py b/pl_bolts/models/rl/dqn_model.py index a4b6008789..69e1cb3a55 100644 --- a/pl_bolts/models/rl/dqn_model.py +++ b/pl_bolts/models/rl/dqn_model.py @@ -404,7 +404,8 @@ def add_model_specific_args(arg_parser: argparse.ArgumentParser, ) -> argparse.A return arg_parser - def _use_dp_or_ddp2(self, trainer: Trainer) -> bool: + @staticmethod + def _use_dp_or_ddp2(trainer: Trainer) -> bool: # for backwards compatibility if _PL_GREATER_EQUAL_1_4: return trainer.accelerator_connector.use_dp or trainer.accelerator_connector.use_ddp2 diff --git a/pl_bolts/models/self_supervised/moco/moco2_module.py b/pl_bolts/models/self_supervised/moco/moco2_module.py index 9f8432de02..9cf89ac7c5 100644 --- a/pl_bolts/models/self_supervised/moco/moco2_module.py +++ b/pl_bolts/models/self_supervised/moco/moco2_module.py @@ -335,7 +335,8 @@ def add_model_specific_args(parent_parser): return parser - def _use_ddp_or_ddp2(self, trainer: Trainer) -> bool: + @staticmethod + def _use_ddp_or_ddp2(trainer: Trainer) -> bool: # for backwards compatibility if _PL_GREATER_EQUAL_1_4: return trainer.accelerator_connector.use_ddp or trainer.accelerator_connector.use_ddp2