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	mmsr	2019-08-23 21:42:47 +08:00
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	Allow passthrough discriminator to have passthrough disabled from config	2020-05-19 09:41:16 -06:00
DiscriminatorResnet_arch.py	Fixup upconv for the next attempt!	2020-05-01 19:56:14 -06:00
feature_arch.py	Several things	2020-09-23 11:56:36 -06:00
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	More NSG improvements (v3)	2020-06-29 20:26:51 -06:00
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	Add simplified SPSR architecture	2020-08-03 10:25:37 -06:00
SRResNet_arch.py	mmsr	2019-08-23 21:42:47 +08:00
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