nos.trainers package
nos.trainers.average_metric module
-
class nos.trainers.average_metric.AverageMetric(name: str, fmt: str = ':f')
Bases: object
-
reset()
-
to_dict()
-
update(val: float, n: int = 1)
nos.trainers.trainer module
-
class nos.trainers.trainer.Trainer(operator: Operator, criterion, optimizer, lr_scheduler: LRScheduler = None, max_epochs: int = 1000, batch_size: int = 16, max_n_logs: int = 200, out_dir: Path = None)
Bases: object
-
eval(loader, model, epoch, device)
-
train(loader, model, epoch, device)
nos.trainers.util module
-
nos.trainers.util.save_checkpoint(operator, val_loss, train_loss, epoch, start, batch_size, train_set, val_set, out_dir: Path = None) → Path
Module contents
-
class nos.trainers.Trainer(operator: Operator, criterion, optimizer, lr_scheduler: LRScheduler = None, max_epochs: int = 1000, batch_size: int = 16, max_n_logs: int = 200, out_dir: Path = None)
Bases: object
-
eval(loader, model, epoch, device)
-
train(loader, model, epoch, device)