Check failed: error == cudaSuccess (2 vs. 0) out of memory

[CrispyCrafter]

It seems we have ran out of memory on the GPU.
How do we set the training batch size?

[JurgenKriel]

You can set the batch size in the train_val.prototxt file found in the following path: cdeep3m-1.6.2/model/inception_residual_train_prediction_1fm
I only use the 1fm model, as it is the only one necessary for 2D data.