storage error

[Mahdidrm]

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