| Hyper-parameter | Model 1 | Model 2 |
---|---|---|---|
Autoencoder | Code size | 64 | 128 |
 | Encoder hidden size | 256, 128 | 512, 256, 128 |
 | Decoder hidden size | 256, 128 | 512, 256, 128 |
GAN | Generator hidden layers | 64, 64 | 128, 128, 128, 128 |
 | Discriminator hidden size | 256, 128 | 512, 256, 128 |
 | # of generator/discriminator steps | 2/1 | 3/1 |