search

facebookresearch / dlrm

An implementation of a deep learning recommendation model (DLRM)
MIT License
3.71k stars 825 forks source link

Are there weights trained on the Criteo Kaggle dataset available? #340

Closed gejinchen closed 1 year ago

gejinchen commented 1 year ago

I am trying to run some inference but the given weights were trained on the Criteo Terabyte Dataset, which is too large to be processed efficiently. Therefore, may I ask whether there are weights trained on the smaller Criteo Kaggle dataset somewhere?

mnaumovfb commented 1 year ago

Hello @gejinchen, similar question has been asked and answered in the prior issue https://github.com/facebookresearch/dlrm/issues/334