Closed valentinitnelav closed 2 years ago
I will now test the YOLOv4 pacsp-s weights. I presume those are lighter weights.
This seems to work for now. A job is running without OMM issues so far using YOLOv4 pacsp-s weights.
See commit 2b9cf6aff2299b5fa177150f37104120c1f400fd
I didn't expect this one, but the following setting overload the 11 GB GPUs;
If we find the tiny weights as mentioned here https://github.com/stark-t/PAI/issues/44 then I can give that one a try. For now, I will move to the V100 GPUs and see if the same happens.
Could it be that YOLOv4 does some cashing in the GPU RAM, or is the architecture simply more demanding than YOLOv5 nano and small? I checked and I do not use the option
--cache-images
.