KeyangXu

icon company Columbia University icon location New York City icon location Ph.D candidate at Columbia University.

Results 2 comments of KeyangXu

PyTorch实现似乎比原paper的结果好？

是的，520k 的 training json，先跑了gen_data.py。 我主要是P@K比较高。 你optimization的参数是什么？

PyTorch实现似乎比原paper的结果好？

> > 是的，520k 的 training json，先跑了gen_data.py。 > > 我主要是P@K比较高。 > > 你optimization的参数是什么？ > > 你好，我是按照代码中的参数进行优化，但是和论文中不同的是代码中的学习率是0.5，而论文中标的是0.01. 请问你是怎么处理的呢 试试 SGD 用 lr =0. 1 然后可以把weight_decay设为0看看。