config.json

{
	"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
}