James Betker 8ab595e427 Add FlatProcessorNet ... After doing some thinking and reading on the subject, it occurred to me that I was treating the generator like a discriminator by focusing the network complexity at the feature levels. It makes far more sense to process each conv level equally for the generator, hence the FlatProcessorNet in this commit. This network borrows some of the residual pass-through logic from RRDB which makes the gradient path exceptionally short for pretty much all model parameters and can be trained in O1 optimization mode without overflows again.		2020-04-28 11:49:21 -06:00
..
archs	Add FlatProcessorNet	2020-04-28 11:49:21 -06:00
__init__.py	Implement downsample GAN	2020-04-24 00:00:46 -06:00
base_model.py	Enable AMP optimizations & write sample train images to folder.	2020-04-21 16:28:06 -06:00
loss.py	mmsr	2019-08-23 21:42:47 +08:00
lr_scheduler.py	mmsr	2019-08-23 21:42:47 +08:00
networks.py	Add FlatProcessorNet	2020-04-28 11:49:21 -06:00
SR_model.py	mmsr	2019-08-23 21:42:47 +08:00
SRGAN_model.py	Config changes for discriminator advantage run	2020-04-25 11:24:28 -06:00
Video_base_model.py	mmsr	2019-08-23 21:42:47 +08:00