search

muslll / neosr

neosr is a framework for training real-world single-image super-resolution networks.
https://github.com/muslll/neosr
Apache License 2.0
136 stars 28 forks source link

Single gradscaler #50

Closed terrainer closed 6 months ago

terrainer commented 6 months ago

The generator and discriminator do not need to be using separate GradScalers. The gradients are cleared after the generator and discriminator steps, meaning there is no reason to separate them.

muslll commented 6 months ago

Good PR :+1: