search

AlexHex7 / DRML_pytorch

Implementation of Deep Region and Multi-Label Learning for Facial Action Unit Detection.
45 stars 14 forks source link

storage error #5

Open Mahdidrm opened 3 years ago

Mahdidrm commented 3 years ago

Hello and thank you very much for sharing this code.

I have a Cuda error in:

                          ```     File: DRML_pytorch-master\lib\region_layer.py", line 44, in forward
                                 grid = self.region_layers[module_name](grid.contiguous()) + grid     ```

The error is:

RuntimeError: CUDA out of memory. Tried to allocate 184.00 MiB (GPU 0; 6.00 GiB total capacity; 3.79 GiB already allocated; 100.36 MiB free; 241.12 MiB cached)

I decreased the batch size in the config file but still get this error. could you please give me some guides?

thanks again and greetings