我正在尝试用tf.Dataset作为我的数据集来拟合Keras模型。我指定了参数steps_per_epoch。input to a model, you should specify the 'steps_per_epoch' argument.此错误使我感到困惑,因为我将steps_per_epoch参数指定为数据集的长度我已经尝试了None以及小于我的数据集长度的整数,但都没有用。下面是我的代码:
def b
我想使用tf.data API。我使用tf.data.from_generator函数来创建迭代器。稍后,我制作了一个可初始化的迭代器。, img) # I train an autoencoder, so the x == y`
dataset = tf.data.Dataset.batch(batch_size).from_generator但是,我甚至试图在我自己的custom_generator中生成批处理大小,但它不起作用。当我想用输入形状