And this is the debug information
terminate called after throwing an instance of ‘c10::Error’
what():  torch::jit::load() received a file from torch.save(), but torch::jit::load() can only load files produced by torch.jit.save()
Exception raised from load at /pytorch/torch/csrc/jit/serialization/import.cpp:347 (most recent call first):
template <typename Value, typename… LoadFromArgs>
void load(Value& value, LoadFromArgs&&… args) {
serialize::InputArchive archive;
archive.load_from(std::forward(args)…);
archive >> value;
}