我有两个数据集
dataset_train = tf.data.Dataset.from_tensor_slices(((x_train, mask_train), y_train))
dataset_dev = tf.data.Dataset.from_tensor_slices(((x_dev, mask_dev), y_dev))
和一个keras模型
model = tf.keras.Model(...)
我想将model
上的dataset_train
放入并在dataset_dev
上进行验证。
model.compile('adam', loss='mse', metrics=['accuracy'])
model.fit(x=dataset_train,
epochs=args.epoch_num,
steps_per_epoch=1,
validation_data=dataset_dev,
validation_steps=1)
我收到以下错误:
ValueError: ('Error when checking model input: expected no data, but got:', (<tf.Tensor 'IteratorGetNext_1:0' shape=(?, 512, 64, 12) dtype=float64>, <tf.Tensor 'IteratorGetNext_1:1' shape=(?, 512, 64, 1) dtype=float64>))
但是当我删除验证数据时,一切正常
model.fit(x=dataset_train,
epochs=args.epoch_num,
steps_per_epoch=1)
知道我在做什么错吗?