spyrit.core.train.train_model_supervised

spyrit.core.train.train_model_supervised(model, criterion, optimizer, scheduler, dataloaders, device, root, num_epochs=25, disp=False, do_checkpoint=0)

Trains the pytorch model in a supervised way