James Betker b980028ca8 Add get_debug_values for vqvae_3_hardswitch		2021-02-03 14:12:24 -07:00
..
__init__.py
kmeans_mask_producer.py
scaled_weight_conv.py
vqvae_3_hardswitch.py	Add get_debug_values for vqvae_3_hardswitch	2021-02-03 14:12:24 -07:00
vqvae_3.py	Register vqvae3	2021-01-29 15:26:28 -07: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	update vqvae to double codebook size for bottom quantizer	2021-01-23 13:47:07 -07:00
vqvae.py