trainerERA5_v1

Attributes

logger

Classes

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

Module Contents

trainerERA5_v1.logger

class trainerERA5_v1.TOADataLoader(conf)

TOA

times_b = None

days_of_year

hours_of_day

__call__(datetime_input)

class trainerERA5_v1.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: int, conf: Dict[str, Any], trainloader: torch.utils.data.DataLoader, optimizer: torch.optim.Optimizer, criterion: torch.nn.Module, scaler: torch.cuda.amp.GradScaler, scheduler: torch.optim.lr_scheduler._LRScheduler, metrics: Dict[str, Any]) → Dict[str, float]

validate(epoch: int, conf: Dict[str, Any], valid_loader: torch.utils.data.DataLoader, criterion: torch.nn.Module, metrics: Dict[str, Any]) → Dict[str, float]