Closed
Description
Conv layer will reshape tensor when batchsize changes. Actually, we should check not only batchsize but also image.width
, image.height
and image.channels
.
This bug will lead to GPU memory error: Cuda Error: an illegal memory access was encountered
.
Aside from image application, it maybe have an impact on ds2.
Metadata
Metadata
Assignees
Labels
No labels