我想使用带有 Then 后端的 Keras 重新训练 VGG 16 的全连接层以获得大灰度图像 (1800x1800)。
所以我有：
- 创建一个具有单一颜色通道的新 VGG，并从原始 VGG 加载权重。
- 将 trainable=False 添加到所有卷积层（根据定义，池化和填充不可训练）
- 删除前两个密集层以仅保留具有两个神经元的输出层
- 大幅增加最大池尺寸和步幅，因为我使用 1800x1800 的输入（别无选择）。尺寸下降很快以匹配原始 VGG 尺寸。
- 减少批量大小以减少所需的内存。
但是当我开始训练时，我遇到了 CNMEM_STATUS_OUT_OF_MEMORY 错误。我使用 NVIDIA K40，所以我有 12Go 的内存。
知道如何解决吗？