| { | |
| "generator": { | |
| "type": "SimpleGenerator", | |
| "layers": [ | |
| { | |
| "type": "Conv2d", | |
| "in_channels": 3, | |
| "out_channels": 64, | |
| "kernel_size": 7, | |
| "stride": 1, | |
| "padding": 3, | |
| "bias": false | |
| }, | |
| { | |
| "type": "BatchNorm2d", | |
| "num_features": 64 | |
| }, | |
| { | |
| "type": "ReLU", | |
| "inplace": true | |
| }, | |
| { | |
| "type": "Conv2d", | |
| "in_channels": 64, | |
| "out_channels": 3, | |
| "kernel_size": 7, | |
| "stride": 1, | |
| "padding": 3, | |
| "bias": false | |
| }, | |
| { | |
| "type": "Tanh" | |
| } | |
| ] | |
| }, | |
| "discriminator": { | |
| "type": "Discriminator", | |
| "layers": [ | |
| { | |
| "type": "Conv2d", | |
| "in_channels": 3, | |
| "out_channels": 64, | |
| "kernel_size": 4, | |
| "stride": 2, | |
| "padding": 1 | |
| }, | |
| { | |
| "type": "LeakyReLU", | |
| "negative_slope": 0.2, | |
| "inplace": true | |
| }, | |
| { | |
| "type": "Conv2d", | |
| "in_channels": 64, | |
| "out_channels": 128, | |
| "kernel_size": 4, | |
| "stride": 2, | |
| "padding": 1 | |
| }, | |
| { | |
| "type": "BatchNorm2d", | |
| "num_features": 128 | |
| }, | |
| { | |
| "type": "LeakyReLU", | |
| "negative_slope": 0.2, | |
| "inplace": true | |
| }, | |
| { | |
| "type": "Conv2d", | |
| "in_channels": 128, | |
| "out_channels": 1, | |
| "kernel_size": 4, | |
| "stride": 1, | |
| "padding": 0 | |
| }, | |
| { | |
| "type": "Sigmoid" | |
| } | |
| ] | |
| } | |
| } | |