Pytorch Examples

Jump to bottom

Haowei Zhang edited this page Mar 1, 2019 · 2 revisions

Welcome to the pytorch-examples wiki!

RankNet train the model from Microsoft Learn To Rank data:

python ranking/RankNet.py

check point is created every 5 epoch. To continue train using previous epoch, use

python ranking/RankNet.py --start_epoch 95 --additional_epoch 5 --lr 0.00002

Clone this wiki locally