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
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
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
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