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
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
networks.py	Fix vgg_gn input_img_factor	2020-08-31 09:50:30 -06:00
SR_model.py
SRGAN_model.py	Enable testing in ExtensibleTrainer, fix it in SRGAN_model	2020-08-31 09:41:48 -06:00