Enable tensorflow v2 compatible code for DIEN

alibaba / ai-matrix

To make it easy to benchmark AI accelerators

Other

179 stars 79 forks source link

Enable tensorflow v2 compatible code for DIEN #69

Closed retonym closed 4 years ago

retonym commented 4 years ago

To run DIEN model with both tensorflow v2 and v1. The code has been tested with tensorflow v1.15 and tensorflow v2.2

CLAassistant commented 4 years ago

Thank you for your submission! We really appreciate it. Like many open source projects, we ask that you sign our Contributor License Agreement before we can accept your contribution.
_{You have signed the CLA already but the status is still pending? Let us recheck it.}

frank-wei commented 4 years ago

@retonym Thanks for changing the code to accommodate TF2.0. Could you create a different folder named DIEN_TF2? I want to keep the original version which still allows people to use TF1.x container to run it. For people interested to run TF2.x, they can use code in DIEN_TF2.

retonym commented 4 years ago

@frank-wei OK. I have moved the TF2 DIEN code to a separate folder.