model

from my_model import MyModel  # import your model class

model = MyModel(...)  # init with same architecture
state_dict = torch.load("model/model.pt", map_location=device)
model.load_state_dict(state_dict)
model.to(device)
model.eval()