Views
No views yet
1import torch 2checkpoint = torch.load('selector.pt') 3model.load_state_dict(checkpoint['model_state_dict'])