Keras：如何计算model.fit（）所需的总内存

Question

每当我点击OOM主板时，它都会挂起，我被迫在Spyder中重新启动内核。

对于2dLocallyConnected网络，如何计算各种批量大小所需的内存？

编辑*代码并未包括在内，因为我希望得到一般答案。参数#4位+批量大小*某些常数。

以下是一个示例网络：

model = Sequential([
LocallyConnected2D(100, kernel_size=(10, 20), strides=1, padding='valid', activation='relu', input_shape=(10, 90, 1)),
 Flatten(),

Dense(1, activation='sigmoid')
])
model.compile(loss='binary_crossentropy', optimizer='adam', metrics=['accuracy'])