Lightrun Answers was designed to reduce the constant googling that comes with debugging 3rd party libraries. It collects links to all the places you might be looking at while hunting down a tough bug.
And, if you’re still stuck at the end, we’re happy to hop on a call to see how we can help out.
Questions about computing resouces
See original GitHub issueHi, thanks for the previous prompt responses.
Today, I ran an experiment based on this repo (u2pl/experiments/pascal/1464/ours/train.sh) with 4 * V100 (32G/gpu) and I got an CUDA OOM error: RuntimeError: CUDA out of memory. Tried to allocate 16.63 GiB (GPU 0; 31.75 GiB total capacity; 5.13 GiB already allocated; 16.63 GiB free; 13.83 GiB reserved in total by PyTorch). This error is thrown out when the program comes into second epoch, i.e., the contrastive loss term is activated.
Since the tutorial in README only mentions the number of GPU but not the capacity, I wonder how many resouces are required for U2PL? Thanks
Issue Analytics
- State:
- Created 2 years ago
- Comments:6 (3 by maintainers)
Top Results From Across the Web
Top Related Medium Post
Top Related StackOverflow Question
Troubleshoot Live Code
Top Related Reddit Thread
Top Related Hackernoon Post
Top Related Tweet
Top Related Dev.to Post
Top Related Hashnode Post
Please feel free to contact me if you have any questions 😃
Hi @Haochen-Wang409, thanks for response. Changing
batch_sizefrom 2 to 1 worked! Thank you very much 😃