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