config.json

{
  "model_type": "heavy_ae",
  "architectures": [
    "HeavyAE"
  ],
  "model_name": "HeavyAE",
  "torch_dtype": "float32",
  "num_channels": 3,
  "image_size": [
    1600,
    720
  ],
  "latent_channels": 8,
  "encoder_layers": 5,
  "decoder_layers": 5,
  "hidden_act": "leaky_relu",
  "out_act": "sigmoid",
  "learning_rate": 0.0001,
  "batch_size": 2,
  "optimizer": "adam",
  "loss_function": "mse",
  "model_path": "model.pt"
}