将 mxnet ImageRecordIter 与 cifar10_val.rec 一起使用时，jupyter notebook 内核死机

Question

单元格如下，正在尝试使用cifar10_val.rec。文件在那里。

def get_data_from_cifar(): 
    train = mx.io.ImageRecordIter() 
    val = mx.io.ImageRecordIter() 
    return train, val 

train, val = get_data_from_cifar() 

每次我 运行 笔记本死机，告诉

The kernel appears to have died. It will restart automatically.

有一个关于 的 post，它是关于内存的，但是它有 to_sparse() 来处理它。

我电脑的内存并不大also.However，而运行在单元格中，taks 管理器不会显示内存耗尽。

这可能是其他问题吗？

更新：

运行 python 解释器中的脚本 shell 正如@leezu 所说

304: [16:36:58] src/io/image_aug_default.cc:282: Check failed: static_cast(res.rows) >= param_.data_shape && static_cast(res.cols) >= param_.data_shape[2] input image size smaller than input shape

我猜我用错了data_shape(3,128,128)。根据cifar10实例更改它们后会更新。

Answer 1

找出cifar10有什么是60000 32*32 images.After把data_shape改成(3,32,32)就不会死了

实际上还必须将 label_width 设置为 1。 但是可以从终端跟踪错误。