search

apache / mxnet

Lightweight, Portable, Flexible Distributed/Mobile Deep Learning with Dynamic, Mutation-aware Dataflow Dep Scheduler; for Python, R, Julia, Scala, Go, Javascript and more
https://mxnet.apache.org
Apache License 2.0
20.75k stars 6.8k forks source link

Issue in custom recordio format. #20802

Open Rushikeshkoli15 opened 2 years ago

Rushikeshkoli15 commented 2 years ago

Description

I am creating a age estimation model and for that I have the image of persons in the age folders(age ranges from 0 to 100) .Image of the person with common age lies in the same folder. In order to train the model on gpu I am using mxnet-cu100. I have created recordio dataset of this data using im2rec.py but while training I got the following error.

Error Message

File "/root/insightface/recognition/arcface_mxnet/verification.py", line 289, in test _embeddings = net_out[0].asnumpy() File "/usr/local/lib/python3.6/dist-packages/mxnet/ndarray/ndarray.py", line 1996, in asnumpy ctypes.c_size_t(data.size))) File "/usr/local/lib/python3.6/dist-packages/mxnet/base.py", line 253, in check_call raise MXNetError(py_str(_LIB.MXGetLastError())) mxnet.base.MXNetError: [18:40:34] include/mxnet/././tensorblob.h:290: Check failed: this->shape.Size() == static_cast(shape.Size()) (5453704704 vs. 1158737408) : TBlob.get_with_shape: new and old shape do not match total elements

plz help and thank you in advance

github-actions[bot] commented 2 years ago

Welcome to Apache MXNet (incubating)! We are on a mission to democratize AI, and we are glad that you are contributing to it by opening this issue. Please make sure to include all the relevant context, and one of the @apache/mxnet-committers will be here shortly. If you are interested in contributing to our project, let us know! Also, be sure to check out our guide on contributing to MXNet and our development guides wiki.