请问必须显存>=10G才能跑吗~

zhougr1993 / DeepInterestNetwork

1.61k stars 559 forks source link

Open Slayerxxx opened 5 years ago

Slayerxxx commented 5 years ago

如题，谢谢

InkdyeHuang commented 4 years ago

算了一下不需要那么显存啊，embeding_size = 8的话embedding参数栈内存只要8M左右

Qiyooo commented 4 years ago

不需要的，如果跑不起来改一下mini_batch就好