search

gusye1234 / LightGCN-PyTorch

The PyTorch implementation of LightGCN
871 stars 230 forks source link

When loading trained model #32

Open theheavystone opened 2 years ago

theheavystone commented 2 years ago

When setting --pretrain as 1 (loading the trained model such as 'lgn-gowalla-3-64.pth.tar'), the initial of lgn model faces the problem: Traceback (most recent call last): File "main.py", line 18, in Recmodel = register.MODELS[world.model_name](world.config, dataset) File "./LightGCN/code/model.py", line 90, in init self.__init_weight() File "./LightGCN/code/model.py", line 112, in __init_weight self.embeddinguser.weight.data.copy(torch.from_numpy(self.config['user_emb'])) KeyError: 'user_emb'

It seems that there is no statement of the config['user_emb'] and config['item_emb'] in world.py or anywhere. Would you kindly like to tell me how to fix this part? How to load the trained model correctly?

Rah1m2 commented 1 year ago

I also have this problem, have you solved it now?