When trainning a model, I got OSError: [Errno 12] Cannot allocate memory
I googled, found the way to set swap file for Nano, but could not find the way to set swap file for NX.
Do you have any suggestions?
Thank you.
More info,
I am training Resnet18.
Here are file counts I used:
Test 6276
Val 6321
Train 93270
The total file size is 7.42 GB
I removed some images, and it works now.
On a computer/server, when I get out of memery error, I set the batch size smaller to fix the issue.
But on jetson NX, decrease batch size do not work… I change it from 32 to 16, but still I got the error.
Should I change it to 8 or 1?
Later, I need to add more images.