标签: tensorflow tensorflow-estimator
对于使用tf.Estimator API创建的模型,是否有一种方法可以测量(或至少估计)数据批加载和向前/向后传递的时间?
到目前为止,我一直在使用tf.train.SessionRunHooks测量模型的性能-但是,至少就我所知,它们只允许我在单次运行估算器{{1 }},train()或evaluate()方法-所有这些方法都涉及在其中加载单批数据。
tf.train.SessionRunHooks
train()
evaluate()