trainerERA5_v2

trainerERA5_v2#

Content:
  • Trainer

    • train_one_epoch

    • validate

    • fit

Attributes#

logger

Classes#

Trainer

Helper class that provides a standard way to create an ABC using

Module Contents#

trainerERA5_v2.logger#
class trainerERA5_v2.Trainer(model: torch.nn.Module, rank: int)#

Bases: credit.trainers.base_trainer.BaseTrainer

Helper class that provides a standard way to create an ABC using inheritance.

train_one_epoch(epoch, conf, trainloader, optimizer, criterion, scaler, scheduler, metrics)#
validate(epoch, conf, valid_loader, criterion, metrics)#
fit_deprecated(conf, train_loader, valid_loader, optimizer, train_criterion, valid_criterion, scaler, scheduler, metrics, rollout_scheduler=None, trial=False)#