(在tensorflow2中)是否比使用自定义训练层(通过使用梯度带)更快地拟合训练模型?

时间:2020-10-19 08:34:10

标签: tensorflow2.0 gradienttape custom-training

我注意到,tensorflow的model.fit训练模型的速度比使用自定义训练循环(通过使用梯度带)训练模型的速度快。如何通过自定义训练循环获得更高的训练速度?

PS-我已经尝试使用@ tf.function

预先感谢:)

0 个答案:

没有答案