Results
2
comments of
Preyum Kumar
I did change batch size in the training script from 4 to 1 but it didn't help. The GPU is 16gb but still the memory runs out. I tried training...
> backend="vanilla", The problem was that only changing the backend from Cuda to vanilla worked, but the epochs are very, very slow.