这不是重复的
在为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>