James Betker 696242064c Use tensor checkpointing to drastically reduce memory usage ... This comes at the expense of computation, but since we can use much larger batches, it results in a net speedup.		2020-09-03 11:33:36 -06:00
..
__init__.py
arch_util.py
AttentionResnet.py
discriminator_vgg_arch.py
DiscriminatorResnet_arch_passthrough.py
DiscriminatorResnet_arch.py
feature_arch.py
FlatProcessorNet_arch.py
FlatProcessorNetNew_arch.py
HighToLowResNet.py
NestedSwitchGenerator.py
ProgressiveSrg_arch.py
ResGen_arch.py
RRDBNet_arch.py
spinenet_arch.py
SPSR_arch.py
SPSR_util.py
SRG1_arch_new.py
SRG1_arch.py
SRResNet_arch.py
SwitchedResidualGenerator_arch.py	Use tensor checkpointing to drastically reduce memory usage	2020-09-03 11:33:36 -06:00