How to use MN3_antispoof.pth.tar

[blackbeandu]

Hello,

i want to use state_dict.

My code :
check_point = torch.load("MN3_antispoof.pth.tar", map_location=torch.device('cpu'))
weight = check_point['state_dict']

and i want to use look like
model = VectorCNN()
model_state_dict = torch.load(weight)
model.load_state_dict(model_state_dict)

"I currently have a class VectorCNN that takes in a .xml and looks for a .bin to load the model in IE format. However, I want to use it with PyTorch itself. I want to save the entire model. For example, 'torch.load(model, 'model.pt').' "

Thanks a lot.