DL-Art-School/codes/models/vqvae
James Betker 587a4f4050 resnet_unet_3
I'm being really lazy here - these nets are not really different from each other
except at which layer they terminate. This one terminates at 2x downsampling,
which is simply indicative of a direction I want to go for testing these pixpro networks.
2021-01-15 14:51:03 -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.py resnet_unet_3 2021-01-15 14:51:03 -07:00
vqvae.py VQVAE 2021-01-07 10:20:15 -07:00
weighted_conv_vqvae.py Enable vqvae to use a switched_conv variant 2021-01-09 20:53:14 -07:00