Feature extraction: CUDA out of memory

[AgarwalVedika]

Hi,

I am trying to extract features for coco dir - however it always gives me RuntimeError: CUDA out of memory.
I have tried using single GPU by setting CUDA_VISIBLE_DEVICES=0- still ti gives me the same Runtime error even when the GPU is free.

What should I do?

Regards,
Vedika

[Cadene]

You should use a smaller batch size. Best luck!

…

_____________________ Remi Cadene PhD Student at LIP6 Sorbonne University http://remicadene.com 0033635504788 Le mer. 22 mai 2019 à 20:32, AgarwalVedika <[email protected]> a écrit :

Hi, I am trying to extract features for coco dir - however it always gives me RuntimeError: CUDA out of memory. I have tried using single GPU by setting CUDA_VISIBLE_DEVICES=0- still ti gives me the same Runtime error even when the GPU is free. What should I do? Regards, Vedika — You are receiving this because you are subscribed to this thread. Reply to this email directly, view it on GitHub <#46?email_source=notifications&email_token=ABDW6LQWVJQVC6XZFNQLCH3PWWGTVA5CNFSM4HOW4ZC2YY3PNVWWK3TUL52HS4DFUVEXG43VMWVGG33NNVSW45C7NFSM4GVI723A>, or mute the thread <https://github.com/notifications/unsubscribe-auth/ABDW6LQFGS7GQYLN4WQBXILPWWGTVANCNFSM4HOW4ZCQ> .