James Betker f5f91850fd hardswitch variant of vqvae3		2021-02-02 21:00:04 -07:00
..
.idea	IDEA update	2020-05-19 09:35:26 -06:00
data	Support training imagenet classifier	2021-01-11 20:09:16 -07:00
models	hardswitch variant of vqvae3	2021-02-02 21:00:04 -07:00
scripts	Hard Routing mods	2021-02-02 20:35:58 -07:00
trainer	Move switched_conv logic around a bit	2021-02-02 20:41:24 -07: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	Support training imagenet classifier	2021-01-11 20:09:16 -07:00
test_image_patch_classifier.py	More refactoring	2020-12-18 09:18:34 -07:00
test.py	misc	2021-01-23 13:45:17 -07:00
train.py	Add switched_conv with hard routing and make vqvae use it.	2021-01-25 08:25:29 -07:00
train2.py	resnet_unet_3	2021-01-15 14:51:03 -07:00