config.json

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