About the checkpoint loading
Opened this issue · 0 comments
kwpoon commented
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:
- "init_ems_embed"->"init_candidate_embed";
- "oneMore"->"project_layer"
- "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.