config.json

{
    "desc": "",
    "accelerator": "auto",
    "devices": 1,
    "abbr": "QM9",
    "mode": "train",
    "desc_ap_file": "../results/desc_ap_file.txt",
    "metric": "validity,qed",
    "z_dim": 32,
    "lambda_gp": 10,
    "lambda_wgan": 0.0,
    "post_method": "softmax",
    "seed": 42,
    "batch_size": 32,
    "num_workers": 8,
    "num_epochs": 150,
    "num_sampled_imgs": 10,
    "dropout_rate": 0.0,
    "n_critic": 5,
    "check_val_every_n_epoch": 5,
    "resume_ckpt_path": null,
    "data_dir": "data/qm9_5k.sparsedataset",
    "log_step": 1,
    "model_save_step": 1,
    "lr_g": 1e-3,
    "lr_d": 1e-3,
    "lr_v": 1e-3,
    "pl_log_name": "lightning_logs",
    "g_conv_dims": [
        128,
        256,
        512
    ],
    "d_conv_dims": [
        [
            128,
            64
        ],
        128,
        [
            128,
            64
        ]
    ],
    "max_epochs": 300,
    "result_dir": "../results/GAN"
}