From d760d52e908599ba742384aa3b3073ad3052d952 Mon Sep 17 00:00:00 2001 From: Younghwan Na Date: Fri, 13 May 2022 15:06:37 +0900 Subject: [PATCH] Fix lint --- mmtune/mm/tasks/mmtrainbase.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/mmtune/mm/tasks/mmtrainbase.py b/mmtune/mm/tasks/mmtrainbase.py index 24ad9585..4d24ef54 100644 --- a/mmtune/mm/tasks/mmtrainbase.py +++ b/mmtune/mm/tasks/mmtrainbase.py @@ -30,7 +30,6 @@ def train_model(self, model: torch.nn.Module, **kwargs) -> None: pass - def contextaware_run(self, status, backend, *args, **kwargs) -> None: from mmtune.mm import hooks # noqa F401 if backend == 'nccl' and os.getenv('NCCL_BLOCKING_WAIT') is None: @@ -52,4 +51,3 @@ def create_trainable(self, backend: str = 'nccl') -> ray.tune.trainable: num_workers=self.args.num_workers, num_gpus_per_worker=self.args.num_cpus_per_worker, num_cpus_per_worker=self.args.num_cpus_per_worker) -