search

qiaoguan / deep-ctr-prediction

CTR prediction models based on deep learning(基于深度学习的广告推荐CTR预估模型)
https://github.com/qiaoguan/deep-ctr-prediction
907 stars 276 forks source link

内存溢出 #10

Closed meilejw closed 4 years ago

meilejw commented 4 years ago

程序一直卡在这个地方，然后占用的内存暴涨，最后退出，请问博主有遇到过类似问题吗？ 20191227163021

qiaoguan commented 4 years ago

把input_fn里面的prefetch 里面的参数调小试试

meilejw commented 4 years ago

内存问题解决了 但是最后预估出来的cvr都是0 会是正样本太少造成的吗？

qiaoguan commented 4 years ago

检查下你的代码逻辑，一般结果可能会很小，很接近0，但一般不会是0

meilejw commented 4 years ago

试了下loss加权求和，cvr不再是0了，应该是cvr的梯度太大把cvr带跑了，但是最终效果不太好 想试下不做loss求和，设置不同的学习率分别优化，用Estimator优化两个loss该怎么实现呢？

qiaoguan commented 4 years ago

没太懂你的意思，完全两个loss没任何关系，不就是两个独立的项目？

meilejw commented 4 years ago

就是ctr和cvr用同一个网络，参数完全共享

qiaoguan commented 4 years ago

如果是完全一起训练，不管怎么做，不管ctr loss和cvr loss怎么定义，最终的loss只有一个