4AI/TDEER

请问下训练时间很慢很慢怎么办，还给了这个警告

Opened this issue · 1 comments

zhangmingdao commented

UserWarning: Converting sparse IndexedSlices to a dense Tensor of unknown shape. This may consume a large amount of memory.
"Converting sparse IndexedSlices to a dense Tensor of unknown shape. "

SeanLee97 commented

请问你使用的 TF 版本是多少呢？
建议使用 tensorflow-gpu==1.15, Keras==2.3.1 并设置 export TF_KERAS=0 确保使用的是 Keras 接口而非 tensorflow.keras 接口