Skip to content

Commit cb179eb

Browse files
committed
disable compiling update_metrics
1 parent 76b00d8 commit cb179eb

File tree

1 file changed

+1
-0
lines changed
  • src/lightning/pytorch/trainer/connectors/logger_connector

1 file changed

+1
-0
lines changed

src/lightning/pytorch/trainer/connectors/logger_connector/result.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -413,6 +413,7 @@ def log(
413413
batch_size = self._extract_batch_size(self[key], batch_size, meta)
414414
self.update_metrics(key, value, batch_size)
415415

416+
@torch.compiler.disable
416417
def update_metrics(self, key: str, value: _VALUE, batch_size: int) -> None:
417418
result_metric = self[key]
418419
# performance: avoid calling `__call__` to avoid the checks in `torch.nn.Module._call_impl`

0 commit comments

Comments
 (0)