使用生成器说明 Tensorboard 中的批处理
Illustrate batch in Tensorboard using a generator
我正在使用 keras 和 fit_generator 来训练我的模型。我正在做的当前模型是一个自动编码器 - 它不会呈现所需的结果。因此，我想创建一个回调，说明每 500 个批次左右的训练图像和真实图像。因此，我想使用 on_batch_begin，但是我不确定如何访问当前批次来创建 tf.Summary.Image。
任何人都可以告诉我有关此的一些信息或知道如何获取当前批次。还是会在生成器中完成？我只是不知道如何附加回调。
我一直没能找到一个优雅的解决方案。我在回调中添加了一个数组，其中包含我要分析的文件。然后我随机选择一张图来说明，用当前的模型在tensorboard上说明。它有效，但我希望它会更优雅 :)
我正在使用 keras 和 fit_generator 来训练我的模型。我正在做的当前模型是一个自动编码器 - 它不会呈现所需的结果。因此，我想创建一个回调，说明每 500 个批次左右的训练图像和真实图像。因此，我想使用 on_batch_begin，但是我不确定如何访问当前批次来创建 tf.Summary.Image。
任何人都可以告诉我有关此的一些信息或知道如何获取当前批次。还是会在生成器中完成？我只是不知道如何附加回调。
我一直没能找到一个优雅的解决方案。我在回调中添加了一个数组，其中包含我要分析的文件。然后我随机选择一张图来说明，用当前的模型在tensorboard上说明。它有效，但我希望它会更优雅 :)