Skip to content
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

About the checkpoint loading #1

Open
kwpoon opened this issue Aug 16, 2023 · 0 comments
Open

About the checkpoint loading #1

kwpoon opened this issue Aug 16, 2023 · 0 comments

Comments

@kwpoon
Copy link

kwpoon commented Aug 16, 2023

Dear authors,

Thank you for the great work! The released codes help a lot in understanding your method!

While I was running the evaluation code, I found some parameter mismatches between the provided checkpoints "blockout_hier_10_ord.pt", "blockout_hier_10_loc.pt" and "blockout_online.pt" and the network DQNBPP.

I'm guessing the parameter names of the checkpoints could be adjusted as follows:

  1. "init_ems_embed"->"init_candidate_embed";
  2. "oneMore"->"project_layer"
  3. "noAction"-> "gat_project_layer"

And also, there seem to be some extra parameters in the checkpoints such that they cannot be loaded directly. I'm wondering if the model loading codes (Line 32 of agent.py) should be modified?

Thanks again, and looking forward to your advice.

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

1 participant