deeplearning4j-examples icon indicating copy to clipboard operation
deeplearning4j-examples copied to clipboard

Expected rank 4 mask array for 2D CNN layers. Mask arrays for 2D CNN layers must have shape [batchSize,channels,X,Y] where X = (1 or activationsHeight) and Y = (1 or activationsWidth): Got rank 2 array with shape [128, 512] Total epochs: 0

Open wswmx opened this issue 6 years ago • 3 comments

Issue Description

Please describe your issue, along with:

  • expected behavior
  • encountered behavior

Version Information

Please indicate relevant versions, including, if relevant:

  • Deeplearning4j version
  • platform information (OS, etc)
  • CUDA version, if used
  • NVIDIA driver version, if in use

Contributing

If you'd like to help us fix the issue by contributing some code, but would like guidance or help in doing so, please mention it!

wswmx avatar Nov 26 '19 02:11 wswmx

我在使用 1.0.0-beta3 版本的时候 用GPU 进行训练出现了 这个问题。 系统是windows
image这个是我的cuda 版本

wswmx avatar Nov 26 '19 02:11 wswmx

image

wswmx avatar Nov 26 '19 02:11 wswmx

Please try again with 1.0.0-beta5.

saudet avatar Dec 02 '19 04:12 saudet