Hao Zhang
It will be out of memory, so use multi-GPU or increase the computational memory of the GPU.
Do you use multi-GPU?
I didn't test in the mobilenetv2.
We use random shape in training, you can see the source code of data processing.
How to add it, can you give me an example @zhanghang1989
It only support Symbol instances.
@chinakook
I used hybridblock to wrap as you say @chinakook
Okay, I will fix that.
Okay, I will fix that.