deeplearning4j / deeplearning4j/deeplearning4j-examples

tf1.4(pb) embedding operator is slow

Open
#1,013 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
Java
Stars
2.5k
Forks
1.8k
PR merge metrics
No merged PRs in 30d

Description

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

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.