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的大小,第二个参数表示某层卷积核的个数,第三个参数表示图像的
相关文章
相关标签/搜索