Open ztqyang opened 1 year ago
Hi, thanks for your code! While Optimizing,it shows RuntimeError. cUDA is out of memory. Is there any way to reduce the required memory.
Hi, you can reduce netchunk_per_gpu. If that is not enough, you could reduce the batch size, however this may also require adapting the learning rate.
Hi, thanks for your code! While Optimizing,it shows RuntimeError. cUDA is out of memory. Is there any way to reduce the required memory.