search

VIPL-SLP / pointlstm-gesture-recognition-pytorch

This repo holds the codes of paper: An Efficient PointLSTM for Point Clouds Based Gesture Recognition (CVPR 2020).
https://openaccess.thecvf.com/content_CVPR_2020/html/Min_An_Efficient_PointLSTM_for_Point_Clouds_Based_Gesture_Recognition_CVPR_2020_paper.html
Apache License 2.0
121 stars 20 forks source link

CUDA out of memory #15

Closed hamza-hadda closed 3 years ago

hamza-hadda commented 3 years ago

When I'm running :python main.py --phase=train --work-dir=PATH_TO_SAVE_RESULTS --device=0 I got this error :

RuntimeError: CUDA out of memory. Tried to allocate 20.00 MiB (GPU 0; 6.00 GiB total capacity; 4.40 GiB already allocated; 0 bytes free; 4.81 GiB reserved in total by PyTorch)

ycmin95 commented 3 years ago

@hamza-hadda You can adopt smaller batch size by changing the config file.