spyrit.core.train.train_model

spyrit.core.train.train_model(model, criterion, optimizer, scheduler, dataloaders, device, root, num_epochs=25, disp=False, do_checkpoint=0, tb_path=False, tb_prof=False, tb_freq=20)

Trains the pytorch model