Closed mbajlo closed 3 years ago
I have made a new dataset with lower number of frames for training and testing the ratio is 5:1 and after that I was able to increase the number of workers to 4, maybe even more, but I kept number 4. I am closing this one.
Hello all,
I am trying to train the network on custom object dataset made by the NDDS. I have a ATI Radeon graphic card so I cannot use the CUDA parallel processing. I have installed
torch
andtorchvision
(both cpu and cp38 versions) and I have tried to run thetrain.py
like this:python train.py --data C:\Users\USER\Desktop\ML\object_dataset\object\ --batchsize 200 --pretrained False --datatest C:\Users\USER\Desktop\ML\object_dataset\object_test\ --object thorHammer --workers 1 --outf outfObject
After the script is running for some time, I got this:
Is there something I can do, beside to get more RAMs? I have win10 with 24GB RAM
any help is appreciated, Thanks