PyTorch Type#

Flytekit PyTorch#

PyTorchCheckpoint([module, hyperparameters, ...])

This class is helpful to save a checkpoint.