我有一个RGB形状的图像,(256,256,3)
并且我有一个形状的权重蒙版(256,256)
。如何使用Keras在它们之间进行逐元素乘法?(所有通道共享相同的掩码)
我正在使用一个对内存要求很高的CNN模型来完成分类任务。这对我在培训期间可以使用的批量大小构成了很大的限制。
一种解决方案是在训练过程中累积梯度,这意味着在每批处理后都不会更新模型的权重。取而代之的是,相同的权重用于几个批次,而每个批次的梯度被累加,然后对单个权重更新操作进行平均。
我正在使用Tensorflow后端Keras,并且我很确定Keras没有现成的功能/方法来实现这一目标。
对于Keras / tensorflow模型,该怎么办?