Skip to content

Log model checkpoints#

Uploading model checkpoints#

You can save model weights from any deep learning framework by using the upload() method.

Model checkpoints appear in the All metadata section. In the below example, they're logged under a field called my_model in the namespace model_checkpoints.

Log PyTorch model weights
my_model = ..., "")

Model checkpoints in all metadata

See example in Neptune 

Tracking checkpoint metadata#

You can also upload the model checkpoints to S3-compatible storage and only track their metadata with Neptune.

For instructions, see Track artifacts.