search

DLR-RM / AugmentedAutoencoder

Official Code: Implicit 3D Orientation Learning for 6D Object Detection from RGB Images
MIT License
340 stars 97 forks source link

The requirement of graphic card memory? #120

Closed 0smile closed 1 year ago

0smile commented 1 year ago

Amazing 6D estimation tools for CAD model, thanks! However, I run into OOM trouble while training my custom model on GTX1650 (4GB memory), even I modified the batch size to 1. Have you tried to train the model on a low video memory device? Your answer is vital for me, thanks!

MartinSmeyer commented 1 year ago

Hey @0smile,

sorry I have not trained on such a small GPU but it should be just enough if it is empty.

Otherwise what you can also do is reduce the size of the encoder/decoder a bit:

https://github.com/DLR-RM/AugmentedAutoencoder/blob/9f0a56f622fabf6200d9f034fcb2eef106997118/auto_pose/ae/cfg/train_template.cfg#L52