Python Keras模型创建占用太多内存？[卡格尔]

我正在使用kaggle和keras构建一个语义分割模型，无论模型的复杂性如何，仅仅构建模型本身就需要15.1GB的GPU空间，因此即使使用生成器，我也没有任何空间来实际加载任何图像。我总是犯错误。我做错什么了吗？为什么keras只需要15GB来创建模型？ 我还没有编译模型。这是在刚刚构建模型之后[

model=model（img\u输入，o）

]

---

您的批处理大小和输入图像形状是什么？@SohaibAnwaar这是在设置批处理大小之前，这是在创建模型之后。默认情况下，TensorFlow分配所有GPU内存并自行管理，因此看到这一点并不奇怪。640x400图像对于这个GPU来说太大了，您必须将图像缩小到更合理的大小。为了进行比较，ImageNet网络根据224x224图像大小进行训练。您的批处理大小和输入图像形状是什么？@SohaibAnwaar这是在设置批处理大小之前，这是在创建模型之后。默认情况下，TensorFlow分配所有GPU内存并自行管理，因此看到这一点并不奇怪。640x400图像对于这个GPU来说太大了，您必须将图像缩小到更合理的大小。为了进行比较，ImageNet网络根据224x224图像大小进行训练。