out of memory about lcdpnet HOT 6 CLOSED

Please try reducing the batch size or image resolution, or add the code to empty the cache periodically.

from lcdpnet.

Thank you for your prompt reply. I adjusted the batchsize to 1 and still got the error. Can you give any more tips on changing the resolution?

from lcdpnet.

Looks like an env error rather than OOM, as it was raised upon import. Please make sure your env has no problem.

from lcdpnet.

I installed torch first: pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
Then proceed to install the libraries in the requirements.txt file. Not sure what's wrong with the environment.

from lcdpnet.

I seem to have found the cause of the problem, because I am running under windows and can train by changing dataloader_num_worker: 5 in the configuration file to 0. But there are always warnings appearing.

from lcdpnet.

This is okay, it will not affect training.

from lcdpnet.