credit.trainers.trainerWRF

Attributes

logger

Classes

TrainerWRF

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

Module Contents

credit.trainers.trainerWRF.logger

class credit.trainers.trainerWRF.TrainerWRF(model: torch.nn.Module, rank: int, conf: dict)

Bases: credit.trainers.base_trainer.BaseTrainer

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

train_one_epoch(epoch, trainloader, optimizer, criterion, scaler, scheduler, metrics)

validate(epoch, valid_loader, criterion, metrics)