fix exception raising (#6901)

This commit is contained in:
Luis Perez 2021-04-08 13:23:26 -07:00 committed by GitHub
parent 742c48e994
commit 90e37ba458
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions

View File

@ -207,7 +207,7 @@ class TensorBoardLogger(LightningLoggerBase):
# todo: specify the possible exception
except Exception as ex:
m = f'\n you tried to log {v} which is not currently supported. Try a dict or a scalar/tensor.'
type(ex)(ex.message + m)
raise ValueError(m) from ex
@rank_zero_only
def log_graph(self, model: LightningModule, input_array=None):