DJSW.train

Functions

train_epoch(model, dataloader, optimizer, criterion, ...)
validate(model, dataloader, criterion, device)
train_pipeline(img_dim, train_loader, val_loader, args)
train_model(args)

Module Contents

DJSW.train.train_epoch(model, dataloader, optimizer, criterion, device)

DJSW.train.validate(model, dataloader, criterion, device)

DJSW.train.train_pipeline(img_dim, train_loader, val_loader, args)

DJSW.train.train_model(args)