Skip to content

Logging 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 in Neptune