alexnet = models.alexnet()
model_path = "model_best.pth.tar"
state_dict = torch.load(model_path,map_location='cpu')
alexnet.load_state_dict(state_dict,strict=False)
alexnet.eval()