James Betker 1a2b9fa130 Get rid of old byol net wrapping ... Simplifies and makes this usable with DLAS' multi-gpu trainer		2021-04-27 12:48:34 -06:00
..
.idea
data	Get segformer to a trainable state	2021-04-25 11:45:20 -06:00
models	Get rid of old byol net wrapping	2021-04-27 12:48:34 -06:00
scripts	Add testing capabilities for segformer & contrastive feature	2021-04-27 09:59:50 -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
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	Add testing capabilities for segformer & contrastive feature	2021-04-27 09:59:50 -06:00