注册
登录
深度学习
Tensorflow 2.0 Keras的训练速度比2.0 Estimator低4倍
返回
Tensorflow 2.0 Keras的训练速度比2.0 Estimator低4倍
作者:
张三岁
发布时间:
2024-08-17 03:53:57 (12天前)
转自:
我们最近切换到Keras for TF 2.0,但是当我们将它与2.0版的DNNClassifier Estimator进行比较时,我们使用Keras的速度降低了大约5倍。但我不能为我的生活弄清楚......
收藏
举报
2 条回复
0#
回复此人
木木
|
2019-08-31 10-32
我认为它较慢,因为它没有在图表上执行。为了在TF2中的图形上执行,你需要一个用tf.function装饰器修饰的函数。查看
这个部分
有关如何重构代码的想法。
编辑
登录
后才能参与评论