Add verb to function getter

Lightning-AI · Nov 23, 2020 · d9289fb · d9289fb
1 parent b937117
commit d9289fb
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pytorch_lightning/trainer/connectors/model_connector.py b/pytorch_lightning/trainer/connectors/model_connector.py
@@ -24,7 +24,7 @@ def __init__(self, trainer):
         self.trainer = trainer
 
     def copy_trainer_model_properties(self, model):
-        ref_model = self._reference_model(model)
+        ref_model = self._get_reference_model(model)
 
         automatic_optimization = ref_model.automatic_optimization and self.trainer.train_loop.automatic_optimization
         self.trainer.train_loop.automatic_optimization = automatic_optimization
@@ -46,9 +46,9 @@ def copy_trainer_model_properties(self, model):
             m.local_rank = self.trainer.local_rank
 
     def get_model(self):
-        return self._reference_model(self.trainer.model)
+        return self._get_reference_model(self.trainer.model)
 
-    def _reference_model(self, model):
+    def _get_reference_model(self, model):
         if self.trainer.accelerator_backend:
             return self.trainer.accelerator_backend.get_reference_model(model)
         return model