config.json

{
    "batch_size": 128,
    "lattent_space_size": 1024,
    "seed": 0,
  "generator_params": {
      "feature_hidden_size": 384,
      "n_transformer_layers": 4,
      "output_hidden_dim": 768,
      "mapping_mlp_params": {
          "layers": [],
          "activation": "gelu",
          "dropout_rate": 0.0
      },
      "transformer_params": {
          "n_head": 4,
          "attention_dropout_rate": 0.2,
          "mlp_layers": [],
          "mlp_activation": "relu",
          "mlp_dropout": 0.2
      }
  },
  "discriminator_params": {
      "n_transformer_layers": 4,
      "encoder_params": {
          "patch_size": 8,
          "overlap": 2,
          "dropout_rate": 0.0
      },
      "transformer_params": {
          "n_head": 4,
          "attention_dropout_rate": 0.2,
          "mlp_layers": [],
          "mlp_activation": "relu",
          "mlp_dropout": 0.2
      },
      "mapping_mlp_params": {
          "layers": [],
          "activation": "gelu",
          "dropout_rate": 0.0
      }
  }
}