James Betker 9c0d7288ea Discretization loss attempt		2021-10-04 20:59:21 -06:00
..
__init__.py
gumbel_quantizer.py	Add switchnorm to gumbel_quantizer	2021-09-24 18:49:25 -06: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	Discretization loss attempt	2021-10-04 20:59:21 -06:00