czy36mengfei/tensorflow2_tutorials_chinese

tf2.0训练速度慢于1.x

aimetrics opened this issue · 4 comments

Hi, @czy36mengfei 请问你有和tf1.0对比过训练的速度吗? 我这边看是不管用keras的optimizer来训练还是用GradientTape + tf.function,2.0的速度都显著慢于1.x,大概慢了2.5倍

你那边是跑什么程序?
我这边跑并没有发现这个问题。
下面是我用keras实现的一个text_cnn的文本分类训练情况:
image

image

@czy36mengfei 可以给个分类的2.0例子吗?我自己的1.14快于2.0,比例也差不多是2.5倍