torch_em.trainer.logger_base
1try: 2 from typing import Literal 3except ImportError: 4 from typing_extensions import Literal # type: ignore 5 6 7class TorchEmLogger: 8 def __init__(self, trainer, save_root, **kwargs): 9 self.trainer = trainer 10 self.save_root = save_root 11 12 def log_train(self, step, loss, lr, x, y, prediction, log_gradients=False): 13 raise NotImplementedError 14 15 def log_validation(self, step, metric, loss, x, y, prediction): 16 raise NotImplementedError
class
TorchEmLogger:
8class TorchEmLogger: 9 def __init__(self, trainer, save_root, **kwargs): 10 self.trainer = trainer 11 self.save_root = save_root 12 13 def log_train(self, step, loss, lr, x, y, prediction, log_gradients=False): 14 raise NotImplementedError 15 16 def log_validation(self, step, metric, loss, x, y, prediction): 17 raise NotImplementedError