Fix fstring in TextLoggerHook (#257)

This commit is contained in:
Maxim Bonnaerens 2020-04-29 15:30:26 +02:00 committed by GitHub
parent 45a39dc6a8
commit f97a8b37ac
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -53,7 +53,7 @@ class TextLoggerHook(LoggerHook):
if torch.cuda.is_available():
log_str += f'memory: {log_dict["memory"]}, '
else:
log_str = 'Epoch({log_dict["mode"]}) ' \
log_str = f'Epoch({log_dict["mode"]}) ' \
f'[{log_dict["epoch"] - 1}][{log_dict["iter"]}]\t'
log_items = []
for name, val in log_dict.items():