IR-BPP icon indicating copy to clipboard operation
IR-BPP copied to clipboard

About the checkpoint loading

Open kwpoon opened this issue 10 months ago • 0 comments

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.

kwpoon avatar Aug 16 '23 09:08 kwpoon