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.

What you logWhat you get

Example

# Log PyTorch model weights
my_model = ...
torch.save(my_model, "my_model.pt")
run["model_checkpoints/my_model"].upload("model_checkpoints/my_model.pt")

See in Neptune