James Betker f129eaa39e Clean up byol a bit ... - Remove option to aug in dataset (there's really no reason for this now that kornia works on GPU on windows) - Other stufff		2021-05-24 21:35:46 -06:00
..
.idea	IDEA update	2020-05-19 09:35:26 -06:00
data	Add zipfilesdataset	2021-05-24 21:35:00 -06:00
models	Clean up byol a bit	2021-05-24 21:35:46 -06:00
scripts	Clean up byol a bit	2021-05-24 21:35:46 -06:00
trainer	Add testing capabilities for segformer & contrastive feature	2021-04-27 09:59:50 -06:00
utils	update vqvae to double codebook size for bottom quantizer	2021-01-23 13:47:07 -07:00
multi_modal_train.py	More adjustments to support distributed training with teco & on multi_modal_train	2020-10-27 20:58:03 -06:00
process_video.py	misc	2021-01-23 13:45:17 -07:00
requirements.txt	Fix some bugs, checkin work on vqvae3	2021-03-02 20:56:19 -07:00
test_image_patch_classifier.py	More refactoring	2020-12-18 09:18:34 -07:00
test.py	misc adjustments for stylegan	2021-04-21 18:14:17 -06:00
train.py	Clean up byol a bit	2021-05-24 21:35:46 -06:00