| { |
| "model_type": "CycleGAN", |
| "checkpoint_file": "checkpoint.pt", |
| "checkpoint_keys": { |
| "model_state_dict": "model_state_dict" |
| }, |
| "domains": { |
| "A": "apple", |
| "B": "orange" |
| }, |
| "directions": { |
| "A_to_B": "apple_to_orange", |
| "B_to_A": "orange_to_apple" |
| }, |
| "image": { |
| "image_size": 256, |
| "channels": 3 |
| }, |
| "preprocessing": { |
| "resize": null, |
| "center_crop": null, |
| "to_tensor": true |
| }, |
| "normalization": { |
| "mean_A": [0.5921, 0.4201, 0.3659], |
| "std_A": [0.2339, 0.2732, 0.2590], |
| "mean_B": [0.6526, 0.4834, 0.2991], |
| "std_B": [0.2155, 0.2118, 0.2252] |
| }, |
| "postprocessing": { |
| "denormalize_A": { |
| "mean": [0.5921, 0.4201, 0.3659], |
| "std": [0.2339, 0.2732, 0.2590] |
| }, |
| "denormalize_B": { |
| "mean": [0.6526, 0.4834, 0.2991], |
| "std": [0.2155, 0.2118, 0.2252] |
| } |
| }, |
| "model_params": { |
| "gen_d_base": 64, |
| "gen_n_down": 2, |
| "gen_n_blocks": 9, |
| "dis_d_in": 3, |
| "dis_d_base": 64, |
| "dis_n_down": 3 |
| } |
| } |
