DL-Art-School/codes
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
..
.idea IDEA update 2020-05-19 09:35:26 -06:00
data Fix LQGT dataset 2020-08-25 11:57:25 -06:00
data_scripts Fix LQGT_dataset, add full_image_dataset 2020-08-24 17:12:43 -06:00
metrics mmsr 2019-08-23 21:42:47 +08:00
models Use tensor checkpointing to drastically reduce memory usage 2020-09-03 11:33:36 -06:00
options More ExtensibleTrainer work 2020-08-23 17:22:45 -06:00
scripts mmsr 2019-08-23 21:42:47 +08:00
temp Misc changes 2020-05-14 20:45:38 -06:00
utils More ExtensibleTrainer work 2020-08-23 17:22:45 -06:00
process_video.py Allow initial temperature to be specified to SPSR net for inference 2020-08-20 11:57:34 -06:00
recover_tensorboard_log.py Misc 2020-07-06 20:44:07 -06:00
requirements.txt Update requirements 2020-08-03 16:57:56 -06:00
run_scripts.sh mmsr 2019-08-23 21:42:47 +08:00
test.py Enable testing in ExtensibleTrainer, fix it in SRGAN_model 2020-08-31 09:41:48 -06:00
train.py Misc fixes & adjustments 2020-09-01 07:58:11 -06:00
train2.py Enable testing in ExtensibleTrainer, fix it in SRGAN_model 2020-08-31 09:41:48 -06:00