trainer_multistep_deprecated

Classes

Trainer

Functions

cleanup()
cycle(dl)
accum_log(log, new_logs)

Module Contents

trainer_multistep_deprecated.cleanup()

trainer_multistep_deprecated.cycle(dl)

trainer_multistep_deprecated.accum_log(log, new_logs)

class trainer_multistep_deprecated.Trainer(model, rank, module=False)

model

rank

device

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

validate(epoch, conf, valid_loader, criterion, metrics)

fit(conf, train_loader, valid_loader, optimizer, train_criterion, valid_criterion, scaler, scheduler, metrics, rollout_scheduler=None, trial=False)