Views
No views yet
1import torch
2import numpy as np
3
4# Load the model (you'll need the network architecture)
5checkpoint = torch.load("model.pt", map_location='cpu')
6
7# The model can be used with the Pyramids environment
8# See the repository for complete usage instructions