James Betker f2a31702b5 Clean stuff up, move more things into arch_util		2021-10-20 21:19:25 -06:00
..
__init__.py	VQVAE	2021-01-07 10:20:15 -07:00
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.py	Add norming to discretization_loss	2021-10-06 17:10:50 -06:00