{
  "architectures": [
    "WGAN_GP"
  ],
  "auto_map": {
    "AutoConfig": "config.Config",
    "AutoModel": "model.WGAN_GP"
  },
  "batch_size": 256,
  "cfg": {
    "batch_size": 256,
    "comet_project": "wgan-gp",
    "comet_workspace": "wgan",
    "d_iter_per_g": 5,
    "data_dir": "/home/user/PyTorch-Lightning-GAN/GAN/celeba",
    "device": "cuda",
    "epochs": 500,
    "final_activation": {
      "g": "tanh"
    },
    "images_dir": "images",
    "img_ch": 3,
    "imsize": 64,
    "lr": {
      "d": 0.0001,
      "g": 0.0001
    },
    "name": "WGAN-GP",
    "norm_type": {
      "d": "GroupNorm",
      "g": "GroupNorm"
    },
    "num_sample_epochs": 20,
    "num_workers": 4,
    "resume": false,
    "samples_dir": "samples",
    "save_image_freq": 50,
    "save_model_freq": 100,
    "seed": 42,
    "use_fabric": false,
    "w_gp": 10,
    "weights_dir": "chkpt",
    "zdim": 128
  },
  "comet_project": "wgan-gp",
  "comet_workspace": "wgan",
  "d_iter_per_g": 5,
  "data_dir": "/home/user/PyTorch-Lightning-GAN/GAN/celeba",
  "device": "cuda",
  "epochs": 500,
  "final_activation": {
    "g": "tanh"
  },
  "images_dir": "images",
  "img_ch": 3,
  "imsize": 64,
  "lr": {
    "d": 0.0001,
    "g": 0.0001
  },
  "model_type": "WGAN_GP",
  "name": "WGAN-GP",
  "norm_type": {
    "d": "GroupNorm",
    "g": "GroupNorm"
  },
  "num_sample_epochs": 20,
  "num_workers": 4,
  "resume": false,
  "samples_dir": "samples",
  "save_image_freq": 50,
  "save_model_freq": 100,
  "seed": 42,
  "torch_dtype": "float32",
  "transformers_version": "4.35.2",
  "use_fabric": false,
  "w_gp": 10,
  "weights_dir": "chkpt",
  "zdim": 128
}