我在张量流中使用估计器框架。我正在输入图像数据集
images.shuffle(buffer_size=3000).batch(100).repeat()
进入火车功能
estimator.train(input_fn=train_input_fn, steps=2000)
因此,我的批次大小为100,并且希望执行2000步训练。一切都会按预期进行,但是只要我添加此代码块
images = tf.image.crop_and_resize(
params["images"],
boxes = np.random.rand(100, 4),
box_ind = [x for x in range(100)],
crop_size = (model_height, model_width)
)
对于我的模型功能,训练在所有图像处理一次后停止,而不是在2000个步骤之后停止。我不明白为什么?