DL-Art-School/codes/models/vqvae
James Betker c0f61a2e15 Rework how DVAE tokens are ordered
It might make more sense to have top tokens, then bottom tokens
with top tokens having different discretized values.
2021-08-05 07:07:17 -06:00
..
__init__.py
kmeans_mask_producer.py Support training imagenet classifier 2021-01-11 20:09:16 -07:00
scaled_weight_conv.py Support training imagenet classifier 2021-01-11 20:09:16 -07:00
vqvae_3_hardswitch.py Fix some bugs, checkin work on vqvae3 2021-03-02 20:56:19 -07:00
vqvae_3_separated_coupler.py Various fixes 2021-07-14 00:08:42 -06:00
vqvae_3.py Various fixes 2021-07-14 00:08:42 -06:00
vqvae_audio_xformer.py Additional work for audio xformer (which doesnt really do a great job) 2021-07-23 10:58:14 -06:00
vqvae_no_conv_transpose_hardswitched_lambda.py Move switched_conv logic around a bit 2021-02-02 20:41:24 -07:00
vqvae_no_conv_transpose_switched_lambda.py Move switched_conv logic around a bit 2021-02-02 20:41:24 -07:00
vqvae_no_conv_transpose.py Various fixes 2021-07-14 00:08:42 -06:00
vqvae.py Rework how DVAE tokens are ordered 2021-08-05 07:07:17 -06:00