James Betker 93a3302819 Push training_state data to CPU memory before saving it ... For whatever reason, keeping this on GPU memory just doesn't work. When you load it, it consumes a large amount of GPU memory and that utilization doesn't go away. Saving to CPU should fix this.		2022-03-04 17:57:33 -07:00
..
.idea
data
models
scripts
trainer	Push training_state data to CPU memory before saving it	2022-03-04 17:57:33 -07:00
utils	Push training_state data to CPU memory before saving it	2022-03-04 17:57:33 -07:00
multi_modal_train.py
process_video.py
requirements.txt
sweep.py
test.py
train.py	Move log consensus to train for efficiency	2022-03-04 13:41:32 -07:00
use_discriminator_as_filter.py