James Betker 0dca36946f Hard Routing mods ... - Turns out my custom convolution was RIDDLED with backwards bugs, which is why the existing implementation wasn't working so well. - Implements the switch logic from both Mixture of Experts and Switch Transformers for testing purposes.		2021-02-02 20:35:58 -07:00
..
.idea
data
models	Hard Routing mods	2021-02-02 20:35:58 -07:00
scripts	Hard Routing mods	2021-02-02 20:35:58 -07:00
trainer	Hard Routing mods	2021-02-02 20:35:58 -07: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
test_image_patch_classifier.py
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