spyrit.core.train.checkpoint

spyrit.core.train.checkpoint(root, epoch, model)[source]

Saves the dictionaries of a given pytorch model for the right epoch