diff --git a/bootstrap/lib/logger.py b/bootstrap/lib/logger.py index 3af4661..ac3ea5d 100755 --- a/bootstrap/lib/logger.py +++ b/bootstrap/lib/logger.py @@ -15,6 +15,7 @@ import inspect import datetime import collections +import torch class Logger(object): @@ -165,7 +166,8 @@ def log_message(self, *message, log_level=INFO, break_line=True, print_header=Tr def log_value(self, name, value, stack_displacement=2, should_print=False, log_level=SUMMARY): if log_level < self.log_level: return -1 - + if type(value) == torch.Tensor: + value = value.item() if name not in self.values: self.values[name] = [] self.values[name].append(value)