Skip to content
This repository has been archived by the owner on Oct 17, 2023. It is now read-only.

When loading trained model #32

Open
theheavystone opened this issue Nov 24, 2022 · 1 comment
Open

When loading trained model #32

theheavystone opened this issue Nov 24, 2022 · 1 comment

Comments

@theheavystone
Copy link

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.embedding_user.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
Copy link

Rah1m2 commented Oct 4, 2023

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

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants