| { | |
| "model_type" : "VQ-GAN_3D", | |
| "batch_size": 1, | |
| "lr": 3e-4, | |
| "n_codes": 16384, | |
| "embedding_dim": 256, | |
| "n_hiddens": 16, | |
| "downsample": [4,4,4], | |
| "discriminator_iter_start": 450, | |
| "perceptual_weight": 4, | |
| "image_gan_weight": 1, | |
| "gan_feat_weight": 4, | |
| "gradient_clip_val": 1.0, | |
| "accumulate_grad_batches": 1, | |
| "internal_resolution": 128, | |
| "max_epochs": 100 | |
| } |