Is there DIEN pretrained model provided?

alibaba / ai-matrix

To make it easy to benchmark AI accelerators

Other

179 stars 79 forks source link

Is there DIEN pretrained model provided? #56

Open hangxu124 opened 4 years ago

hangxu124 commented 4 years ago

Hi,

I'd like to know if there is pretrained model for DIEN provided? It seems that currently in the dnn_best_model_trained, only .meta and .index file are provided. Could you please upload the .data file for this checkpoint including weights bias gradients?

Thanks

rabbitwayne commented 4 years ago

You can get the data file in https://zenodo.org/record/3463683/files/ckpt_noshuffDIEN3.data-00000-of-00001 See prepare_dataset.sh for details.