Unable to find ffnn.pkl file

nzc / dnn_ctr

The framework to deal with ctr problem。The project contains FNN,PNN,DEEPFM, NFM etc

756 stars 285 forks source link

Unable to find ffnn.pkl file #14

Open lthakur007 opened 5 years ago

lthakur007 commented 5 years ago

Hi, Could you please help me fix the following issue. Traceback (most recent call last): File "main.py", line 4, in from model import FNN File "/root/dnn-ctr-fnn/dnn_ctr/model/FNN.py", line 540, in fnn.load_state_dict(torch.load('./data/model/ffnn.pkl')) File "/root/.local/lib/python2.7/site-packages/torch/serialization.py", line 356, in load f = open(f, 'rb') IOError: [Errno 2] No such file or directory: './data/model/ffnn.pkl'

nzc commented 5 years ago

@luckynikki This is my fault. You are supposed to Document the code in line 540 (fnn.load_state_dict(torch.load('./data/model/ffnn.pkl'))) and use the code in line 541 to pretrain the fm model and save it.

lthakur007 commented 5 years ago

Thank you for the help. It worked :).