DL-Art-School/codes
James Betker 15e00e9014 Finish integration with autocast
Note: autocast is broken when also using checkpoint(). Overcome this by modifying
torch's checkpoint() function in place to also use autocast.
2020-10-22 14:39:19 -06:00
..
.idea IDEA update 2020-05-19 09:35:26 -06:00
data Misc nonfunctional mods to datasets 2020-10-22 10:16:17 -06:00
metrics mmsr 2019-08-23 21:42:47 +08:00
models Finish integration with autocast 2020-10-22 14:39:19 -06:00
scripts Misc 2020-10-21 11:08:21 -06:00
switched_conv@cb520afd4d Update SwitchedConv 2020-10-13 20:57:12 -06:00
utils Codebase cleanup 2020-10-13 20:56:39 -06:00
multi_modal_train.py Add multi-modal trainer 2020-10-22 13:27:32 -06:00
process_video.py Fix process_video 2020-10-21 11:08:12 -06:00
requirements.txt Add missing requirements 2020-09-09 10:46:08 -06:00
test.py Restore test.py for use as standalone validator 2020-10-19 15:26:07 -06:00
train.py Add multi-modal trainer 2020-10-22 13:27:32 -06:00
train2.py Add multi-modal trainer 2020-10-22 13:27:32 -06:00