Pretraining details

Question

Pretraining details

YH-UtMSB opened this issue 3 years ago · 3 comments

Hi Peijie, thank you for open sourcing the models. I am trying to re-implement diffnet in pytorch and it would be a great help if you could let me know the details of model pretraining. Specifically, how was the model by the name "diffnet_hr_0.3437_ndcg_0.2092_epoch_98.ckpt" trained? Thanks in advance.

Answer 1 · 2022-02-18T02:07:48.000Z

The only thing you need is to run the code without any pre-training model, saving the model when the model achieves the largest NDCG value.

Answer 2 · 2022-02-18T02:09:24.000Z

And you can refer to the paddlepaddle version of our model at this link https://gitee.com/peijie_hfut/diffnet-paddlepaddle. The paddlepaddle is similar to the pytorch.

Answer 3 · 2022-02-20T20:22:20.000Z

Thanks for the reply! This is very helpful.