James Betker
|
515905e904
|
Add a min_loss that is DDP compatible
|
2020-10-28 15:46:59 -06:00 |
|
James Betker
|
61e5047c60
|
Fix loss accumulator when buffers are not filled
They were reporting incorrect losses.
|
2020-10-09 19:47:59 -06:00 |
|
James Betker
|
dd9d7b27ac
|
Add more sophisticated mechanism for balancing GAN losses
|
2020-10-02 22:53:42 -06:00 |
|
James Betker
|
dffc15184d
|
More ExtensibleTrainer work
It runs now, just need to debug it to reach performance parity with SRGAN. Sweet.
|
2020-08-23 17:22:45 -06:00 |
|
James Betker
|
f40545f235
|
ExtensibleTrainer work
|
2020-08-22 08:24:34 -06:00 |
|