使用pycharm加载model_name.pth可能会报错,这可能是由于jupyter和pycharm的解析造成的,如果是使用jupyter请使用上述代码,如果是pycharm,更改为下面的即可。
model_weight_path = "model_name.pth"
model1 = torch.load(model_weight_path)
使用pycharm加载model_name.pth可能会报错,这可能是由于jupyter和pycharm的解析造成的,如果是使用jupyter请使用上述代码,如果是pycharm,更改为下面的即可。
model_weight_path = "model_name.pth"
model1 = torch.load(model_weight_path)