deeplearning4j / deeplearning4j-examples

Deeplearning4j Examples (DL4J, DL4J Spark, DataVec)

Home Page:http://deeplearning4j.konduit.ai

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

tf1.4(pb) embedding operator is slow

walegahaha opened this issue · comments

Hi,

If I use tf1.4 model with pb format, it takes almost 200ms in DeepFM (70 features, 3 dim each embedding lookup vector). After debug, we catch the problems in embedding operator