1. 程式人生 > >ResourceExhaustedError: OOM when allocating tensor with shape[10,256,400,528]

ResourceExhaustedError: OOM when allocating tensor with shape[10,256,400,528]

使用keras庫進行模型訓練時,出現以下錯誤:

ResourceExhaustedError (see above for traceback): OOM when allocating tensor with shape[10,256,400,528]

解決方法如下:

[10,256,400,528]的第一個引數表示batch_size的大小,第二個引數表示某層卷積核的個數,第三個引數表示影象的高,第四個引數表示影象的長
這裡出現這種錯誤的原因時超出記憶體了,因此可以適當減小batch_size的大小即可解決