| { |
| "architectures": [ |
| "WGANGP_generator" |
| ], |
| "auto_map": { |
| "AutoConfig": "config.Config", |
| "AutoModel": "model.WGANGP_generator" |
| }, |
| "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": "WGANGP_generator", |
| "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": "WGANGP_generator", |
| "name": "WGANGP_generator", |
| "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 |
| } |
|
|
