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的大小,第二個參數表示某層卷積核的個數,第三個參數表示圖像的
相關文章
相關標籤/搜索