SpA-Former-shadow-removal icon indicating copy to clipboard operation
SpA-Former-shadow-removal copied to clipboard

Published 20 hours ago • verified publisher icon zhangbaijin

configuration and memory?

Open LiliiiH opened this issue 1 year ago • 9 comments

Thanks for your great work! But I can't find any instructions for configuration, can you tell me how to set up the environment in order to run predict.py. Thanks!

LiliiiH avatar Jul 29 '22 11:07 LiliiiH

python3.7-3.8 pytorch1.4-1.8. It's ok

zhangbaijin avatar Jul 30 '22 00:07 zhangbaijin

Thank you so much!

LiliiiH avatar Jul 30 '22 14:07 LiliiiH

I met this error when I wanted to run train.py: RuntimeError: CUDA out of memory. Tried to allocate 400.00 MiB (GPU 0; 24.00 GiB total capacity; 18.64 GiB already allocated; 2.32 GiB free; 19.11 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation. See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF Can you tell me how to solve it? Is 'setting max_split_size_mb' the only way to solve? Thanks for your patience!

LiliiiH avatar Jul 30 '22 15:07 LiliiiH

Sorry for late to reply you, have you solved the Our of memory problem?

zhangbaijin avatar Sep 01 '22 04:09 zhangbaijin

I am facing the same, "CUDA out of memory" issue, Is there any solution

HumeraSabir avatar Sep 29 '22 16:09 HumeraSabir

I am facing the same, "CUDA out of memory" issue,

Mnurr avatar Apr 21 '23 07:04 Mnurr

我是在单张3090训练和推理的，我建议你们可以通过big_resize.py将输入变为256x256，这样应该不会报out of memory的错误

zhangbaijin avatar Nov 15 '23 03:11 zhangbaijin

By lowering the batch size and changing the input image size to 256x256, I was able to train on a 3060ti with 8GB memory.👍👍

ShunTatsukawa avatar Nov 22 '23 02:11 ShunTatsukawa

It's really happy to help you, if you have any questions, please email me.

zhangbaijin avatar Nov 22 '23 03:11 zhangbaijin