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
..
archs	Use tensor checkpointing to drastically reduce memory usage	2020-09-03 11:33:36 -06:00
steps	Add InterpolateInjector	2020-09-03 11:32:47 -06:00
__init__.py	ExtensibleTrainer work	2020-08-22 08:24:34 -06:00
base_model.py	Fix multistep optimizer (feeding from wrong config params)	2020-08-04 16:42:58 -06:00
ExtensibleTrainer.py	Misc fixes & adjustments	2020-09-01 07:58:11 -06:00
feature_model.py	feamod fix	2020-08-30 08:08:49 -06:00
loss.py	Update loss with lr crossgan	2020-08-26 17:57:22 -06:00
lr_scheduler.py	Extensible trainer (in progress)	2020-08-12 08:45:23 -06:00
networks.py	Fix vgg_gn input_img_factor	2020-08-31 09:50:30 -06:00
SR_model.py	Allow validating in batches, remove val size limit	2020-06-02 08:41:22 -06:00
SRGAN_model.py	Enable testing in ExtensibleTrainer, fix it in SRGAN_model	2020-08-31 09:41:48 -06:00