Skip to content

experiments.py: error: argument --distmult_state_dict_path: expected one argument #4

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
swxhha opened this issue Dec 27, 2022 · 1 comment

Comments

@swxhha
Copy link

swxhha commented Dec 27, 2022

When I train an RL model (policy gradient + reward shaping) on the NELL-995 dataset, I execute the following commands:
./experiment-rs.sh configs/nell-995-rs.sh --train 0 --test
Then I ran into the following problem:
experiments.py: error: argument --distmult_state_dict_path: expected one argument
What is the role of distmult_state_dict_path ? What input does it need?

@kingsaint
Copy link
Owner

kingsaint commented Apr 16, 2023

@swxhha Sorry for the delayed response.
You first need to train a ConvE model. As I've described in the README note To train the RL models using reward shaping, make sure 1) you have pre-trained the embedding-based ConvE model and 2) set the file path pointers conve_state_dict_path to the pre-trained embedding-based models correctly in the configs/<dataset>-rs.sh or configs/<dataset>.sh files.

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

No branches or pull requests

2 participants