Skip to content
Open
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 5 additions & 1 deletion src/lightning/fabric/loggers/tensorboard.py
Original file line number Diff line number Diff line change
Expand Up @@ -211,7 +211,11 @@ def log_metrics(self, metrics: Mapping[str, float], step: Optional[int] = None)
else:
try:
self.experiment.add_scalar(k, v, step)
except (NotImplementedError, ValueError) as ex:
except (
NotImplementedError,
ValueError,
ModuleNotFoundError, # https://github.com/pytorch/pytorch/issues/24175
) as ex:
raise ValueError(
f"\n you tried to log {v} which is currently not supported. Try a dict or a scalar/tensor."
) from ex
Expand Down
Loading