To use this model, you must define the same architecture used during training, then load the weights as follows:
python
1import torch
23model =...# define your model architecture here4model.load_state_dict(torch.load("trained-model.pth", map_location="cpu"))5model.eval()
