How much memory is needed to run a CNN deep learning net with CSV data?

FriedemannFriedemann Member, University Professor Posts: 27 University Professor
edited April 2021 in Help
When running the MNIST example (CSV Version) with the manual input shape setting for "Convolutional Flattened"
(as described in here: using the CUDA backend I get an error message saying: Failed to allocate 1,685,712,896 bytes from HOST memory.

When running with the CPU backend, my machine simply runs out of memory and starts swapping (JVM memory is restricted to 10,000 MB and backend memory to 10G).

I guess that something is very wrong with the FeedForwardToCnnPreProcessor.

Sign In or Register to comment.