James Betker 51b63b2aa6 Add switched_conv with hard routing and make vqvae use it.		2021-01-25 08:25:29 -07:00
..
__init__.py	VQVAE	2021-01-07 10:20:15 -07:00
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_no_conv_transpose_switched_lambda.py	Add switched_conv with hard routing and make vqvae use it.	2021-01-25 08:25:29 -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	VQVAE	2021-01-07 10:20:15 -07:00