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
TorchEmLogger(trainer, save_root, **kwargs) 
 9    def __init__(self, trainer, save_root, **kwargs):
10        self.trainer = trainer
11        self.save_root = save_root
trainer
save_root
def log_train(self, step, loss, lr, x, y, prediction, log_gradients=False): 
13    def log_train(self, step, loss, lr, x, y, prediction, log_gradients=False):
14        raise NotImplementedError
def log_validation(self, step, metric, loss, x, y, prediction): 
16    def log_validation(self, step, metric, loss, x, y, prediction):
17        raise NotImplementedError