Keras-无法在组件0中批量处理具有不同形状的张量

时间:2019-11-18 10:08:20

标签: tensorflow keras batch-processing batching

这不是重复的

在为Keras模型使用数据生成器时出现以下错误。关于该特定错误,有很多关于StackOverflow的文章,但我认为它们都不适用于我的情况(或者至少我无法将根本原因翻译成此问题)

1

当我使用def fetch_model(): new_model = Sequential() new_model.add(BatchNormalization(input_shape=(10, 128))) # The input shape excludes batch new_model.add(Flatten()) new_model.add(Dense(2048, activation="relu")) new_model.add(Dense(audioset_label_count, activation="sigmoid")) new_model.compile("adam", loss="binary_crossentropy", metrics=["accuracy"]) return new_model 的批处理大小时,一切正常,但是对于> 1的批处理大小,一切开始崩溃。很高兴了解这里到底发生了什么?

以下是复制问题的简单模型

python train.py

并且我正在Google的AudioSet数据上训练模型,并且已将其样本添加到https://github.com/i2infinity/python-demo

要重现此问题,请在https://github.com/i2infinity/python-demo.git上运行<option> <i class="flag flag-us"></i> America </option>

0 个答案:

没有答案