提问人：solveForX 提问时间：11/18/2023 更新时间：11/18/2023 访问量：12
使用 tensorflow2 训练模型时识别瓶颈
Identify bottleneck while training model using tensorflow2
问：
我是 Tensorflow 的新手。我想确定我的模型训练的哪一部分需要时间：是数据加载还是前向传递模型。
在 pytorch 中，我只需注释前向传递、损失计算和后向传递，它们在 for 循环下表示 epoch 数，并了解数据加载器花费了多少时间。
但是，在 TF.我几乎被 model.fit.generator 卡住了或碰壁了。
知道如何实现类似的瓶颈分析吗？
答： 暂无答案
评论
ImageDataGeneratorfit_generator