tf2.0训练速度慢于1.x
aimetrics opened this issue · 4 comments
aimetrics commented
Hi, @czy36mengfei 请问你有和tf1.0对比过训练的速度吗? 我这边看是不管用keras的optimizer来训练还是用GradientTape + tf.function，2.0的速度都显著慢于1.x，大概慢了2.5倍
czy36mengfei commented
你那边是跑什么程序？
我这边跑并没有发现这个问题。
下面是我用keras实现的一个text_cnn的文本分类训练情况：
KANGRuipeng commented
@czy36mengfei 可以给个分类的2.0例子吗？我自己的1.14快于2.0，比例也差不多是2.5倍