psquare model is not saved after running `train_psquare.py`

Question

psquare model is not saved after running `train_psquare.py`

OneDirection9 opened this issue 4 years ago · comments

Hi,
great work and thanks for sharing.

When I run train_transmitter.py, train_receiver_model.py, and train_psquare.py step by step. I can get the transmitter and receiver model, except the psquare model. I go through the code and can't fix it by myself. Can you help?

Thanks.

Qian · Answer 1 · Sun Nov 15 2020 22:13:36 GMT+0800 (China Standard Time)

@OneDirection9
Hi, thanks for your interests on our work. Please check if there is any output log matching the pattern Save model at, and you could find the psqure model checkpoint there.

Zhipeng Han · Answer 2 · Mon Nov 16 2020 10:13:43 GMT+0800 (China Standard Time)

Hi @SivilTaram ,

Thanks for your reply.
The default save directory is tmp, I can find the transmitter and receiver model under there. But in the psquare folder, there is only a tensorboard folder.

Qian · Answer 3 · Mon Nov 16 2020 17:20:09 GMT+0800 (China Standard Time)

@OneDirection9 Hi, thanks for your feedback! Since I just checked evaluation scripts carefully before releasing the code (the model weights are the ones saved before submitting the paper to ACL), so it is possible that the current code is not totally compatible with the provided ParlAI version. And I will try to reproduce the result as soon as possible and get back here in 2 days. Please stay tuned ☕

Qian · Answer 4 · Tue Nov 17 2020 00:40:05 GMT+0800 (China Standard Time)

@OneDirection9 Hi~ Many thanks for your feedback. Yeah it is indeed an issue for our code (sorry for that). And it could work well now, and pls have a try ☕