Hi ptrblck,
Thank you very much for your response.
Here is my code link contains this error (https://github.com/leesays92/NN-in-general/blob/master/Error1.ipynb 4). I would really appreciate it if you could look through it.
I’m currently using GPU for RTX 2080 Ti.
I upgraded to PyTorch 1.4.0 (Which was originally 1.3.1) and it doesn’t work as well.
The funny thing is that… If i reduce the total amount of data (11,7000 to 35,000), it works.
Thanks in advance!