search

rentruewang / koila

Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code.
https://rentruewang.com/koila/
Apache License 2.0
1.82k stars 62 forks source link

Compatibility with GANs? #10

Closed Otje89 closed 2 years ago

Otje89 commented 2 years ago

Not an issue, but a question. Would you think this works well and correctly in a GAN setting where two networks competing with each other?

rentruewang commented 2 years ago

Hi, it should work fine with GANs, with or without gradient reversal layers, provided that it's written in PyTorch.