Swin-Transformer
Swin-Transformer copied to clipboard
About the continued growth of CPU memory!
When I use GPU to train SWIN, the CPU memory keeps increasing, I want to ask what is going on? Is it a code bug? Very distressed this problem, hope to get a reply!
Also facing the same problem here
https://github.com/pytorch/pytorch/issues/13246#issuecomment-905703662 @zhubingerdai Try this one~