From d28e36566957dcfc29536e932f0eb974dc738f4b Mon Sep 17 00:00:00 2001 From: ftorres16 <36959980+ftorres16@users.noreply.github.com> Date: Fri, 13 May 2022 14:22:09 -0400 Subject: [PATCH] Remove unused hook keyword argument (#13059) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-authored-by: Carlos Mocholí --- pytorch_lightning/callbacks/device_stats_monitor.py | 7 +------ 1 file changed, 1 insertion(+), 6 deletions(-) diff --git a/pytorch_lightning/callbacks/device_stats_monitor.py b/pytorch_lightning/callbacks/device_stats_monitor.py index 00fd79d0f7..7b181fe463 100644 --- a/pytorch_lightning/callbacks/device_stats_monitor.py +++ b/pytorch_lightning/callbacks/device_stats_monitor.py @@ -100,12 +100,7 @@ class DeviceStatsMonitor(Callback): logger.log_metrics(prefixed_device_stats, step=trainer.fit_loop.epoch_loop._batches_that_stepped) def on_train_batch_start( - self, - trainer: "pl.Trainer", - pl_module: "pl.LightningModule", - batch: Any, - batch_idx: int, - unused: Optional[int] = 0, + self, trainer: "pl.Trainer", pl_module: "pl.LightningModule", batch: Any, batch_idx: int ) -> None: self._get_and_log_device_stats(trainer, "on_train_batch_start")