Segmetation fault occurs after mnist cpp code in GPU #597
Description
I tried reducing the batch size to 4. It does not throw the error when using CPU. I ran the code with 15 epochs.
{{ message }}
I tried reducing the batch size to 4. It does not throw the error when using CPU. I ran the code with 15 epochs.