replace list for loop with tf bool mask to make it more gpu friendly.

guoday / ctrNet-tool

This's the tool for CTR, including FM, FFM, NFFM and so on.

317 stars 93 forks source link

Closed daxiongshu closed 5 years ago

daxiongshu commented 5 years ago

I tested it locally and it should be correct.

guoday commented 5 years ago

great, it looks less GPU memory.