James Betker d7ee14f721 Move to torch.cuda.amp (not working) ... Running into OOM errors, needs diagnosing. Checkpointing here.		2020-10-22 13:58:05 -06:00
..
panet	Add PANet arch	2020-10-12 10:20:55 -06:00
__init__.py
arch_util.py	Add a ChainedEmbeddingGen which can be simueltaneously used with multiple training paradigms	2020-10-21 22:21:51 -06:00
ChainedEmbeddingGen.py	Move to torch.cuda.amp (not working)	2020-10-22 13:58:05 -06:00
discriminator_vgg_arch.py	Remove separated vgg discriminator	2020-10-18 12:10:24 -06:00
DiscriminatorResnet_arch_passthrough.py
DiscriminatorResnet_arch.py
feature_arch.py
ProgressiveSrg_arch.py	Import switched_conv as a submodule	2020-10-07 23:10:54 -06:00
rcan.py	Allow checkpointing to be disabled in the options file	2020-10-03 11:03:28 -06:00
ResGen_arch.py
RRDBNet_arch.py	Allow checkpointing to be disabled in the options file	2020-10-03 11:03:28 -06:00
spinenet_arch.py	Fix spinenet bug	2020-10-17 20:20:36 -06:00
SPSR_arch.py	Restore test.py for use as standalone validator	2020-10-19 15:26:07 -06:00
SPSR_util.py
SRResNet_arch.py
StructuredSwitchedGenerator.py	SSGSimpler network	2020-10-15 17:18:44 -06:00
SwitchedResidualGenerator_arch.py	Set post_transform_block to None where applicable	2020-10-15 17:20:42 -06:00