Hi,
I created a custom gluon Dataset
where records are sequences of 6 pictures, and trying to feed it to a data loader.
What works:
- I can get
Dataset
records fine usingDataset.__getitem__()
- I can get batches out of the
DataLoader
when usingbatch_size
= 1
What doesn’t work:
- when I use batch_size > 1, eg 2 or more, I have this error:
raise MXNetError(py_str(_LIB.MXGetLastError())) mxnet.base.MXNetError: [15:08:51] include/mxnet/././tensor_blob.h:290: Check failed: this->shape_.Size() == shape.Size() (903168 vs. 752640) TBlob.get_with_shape: new and old shape do not match total elements
mx 1.4.1 on AWS SageMaker EC2 P3.2xl