Open lhy1214 opened 1 year ago
Sorry, I do not quite understand your question. You first need to make a directory named "saved_model", and then run our codes, the checkpoint will saved in the directory automatically. The training procedure would not take so long (generally within 10 epochs), and i believe our codes can achieve better results than other baselines in our paper.
With your help, the problem has been solved, thank you very much again.
Hello, is there no folder and file?' saved_model/net_parameter1.pkl def save_model(self,name=''): if name!='': torch.save(self.state_dict(), 'saved_model/'+name+'.pkl') else: torch.save(self.state_dict(),'saved_model/net_parameter1.pkl')