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	IDEA update	2020-05-19 09:35:26 -06:00
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	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	Add testing capabilities for segformer & contrastive feature	2021-04-27 09:59:50 -06:00