Hello, I would like to train a new model based on your code, but after I trained the pointer network, and I added torch.save(trainer.model, "name.pt") in run_new_training() at ptr_test.py. But I still cannot inference the seq, and it said that there is no decoded_seq in my file. Could you tell me how did you save your trained_model.pt?
Thank you very much!
Hello, I would like to train a new model based on your code, but after I trained the pointer network, and I added torch.save(trainer.model, "name.pt") in run_new_training() at ptr_test.py. But I still cannot inference the seq, and it said that there is no decoded_seq in my file. Could you tell me how did you save your trained_model.pt?
Thank you very much!