Closed Einhartd closed 7 months ago
I have not tried training the network with 4GB of memory. You could try modifying the backbone to Resnet50. Or further minimizing the batch size to 1 (you may need to tune learning rate after this).
But it will be difficult to reproduce the full result.
It worked! Thank you so much!
Dist: Pop OS 22.04 nvidia-smi output:
nvcc --version output:
I have encountered an error while trying to train my first model on YOLO3D model. I have simply followed instruction in docs/mono3d.md After entering this command:
./launchers/train.sh config/CONFIG_FILE_YOLO.py 0 proba
program crashed with error below:I have tried reducing batch_size value from 8 to 2 in file CONFIG_FILE_YOLO.py which was generated by this command:
cp Yolo3D_example $CONFIG_FILE.py
Have you got any idea how to reduce allocated memory ?