search

PaddlePaddle / PaddleRec

Recommendation Algorithm大规模推荐算法库，包含推荐系统经典及最新算法LR、Wide&Deep、DSSM、TDM、MIND、Word2Vec、Bert4Rec、DeepWalk、SSR、AITM，DSIN，SIGN，IPREC、GRU4Rec、Youtube_dnn、NCF、GNN、FM、FFM、DeepFM、DCN、DIN、DIEN、DLRM、MMOE、PLE、ESMM、ESCMM, MAML、xDeepFM、DeepFEFM、NFM、AFM、RALM、DMR、GateNet、NAML、DIFM、Deep Crossing、PNN、BST、AutoInt、FGCNN、FLEN、Fibinet、ListWise、DeepRec、ENSFM，TiSAS，AutoFIS等，包含经典推荐系统数据集criteo 、movielens等
https://paddlerec.readthedocs.io/
Apache License 2.0
4.16k stars 718 forks source link

drop_last=False not working #539

Open simwiki opened 2 years ago

simwiki commented 2 years ago

https://github.com/PaddlePaddle/PaddleRec/blob/df31a2d9c63e900039a28e2f278de0c51e19570f/tools/paddle_infer.py#L112

yinhaofeng commented 2 years ago

将这个设置改为False后，会导致因为数据集样本数不能被 batch_size 整除而产生的最后一个不完整的mini-batch。所以我们推荐使用True。或者在您的模型中检查batch_size如果不符合预期时发生的情况