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
data Misc nonfunctional mods to datasets 2020-10-22 10:16:17 -06:00
metrics
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
utils
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
test.py
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