Keras: 计算 model.fit() 所需的总内存

Question

每次我碰到OOM床头板，它就会挂起，我不得不重启Spyder中的内核。在

对于2dLocalConnected网络，如何计算不同批处理大小所需的内存？在

编辑代码不包括在内，因为我希望得到一个普遍的答案。参数的长度是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'])