tinymind_competition icon indicating copy to clipboard operation
tinymind_competition copied to clipboard

Published 20 hours ago • verified publisher icon 610265158

训练错误

Open icsaas opened this issue 6 years ago • 1 comments

运行ｂａｓｈ ./train.sh ,使用resnext-50模型，出现错误：

9, in check_call raise MXNetError(py_str(_LIB.MXGetLastError())) mxnet.base.MXNetError: [13:38:02] src/storage/./pooled_storage_manager.h:108: cudaMalloc failed: out of memory

icsaas avatar May 06 '18 05:05 icsaas

显存不够, 减小batch_size,或者减小data_shape, 效果可能会有变化，: ）

610265158 avatar May 06 '18 06:05 610265158