search

WindyLee0822 / TREA

Source code of “TREA: Tree-structure Reasoning Schema for Conversational Recommendation (ACL 2023)”
4 stars 1 forks source link

RuntimeError: Parent directory saved_model does not exist. #4

Open lhy1214 opened 1 year ago

lhy1214 commented 1 year ago

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')

WindyLee0822 commented 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.

lhy1214 commented 1 year ago

With your help, the problem has been solved, thank you very much again.