{
  "act_layer": "identity",
  "architectures": [
    "ACTUForImageSegmentation"
  ],
  "backbone": "convnextv2_base",
  "batch_first": true,
  "bias": true,
  "bidirectional": false,
  "climate": false,
  "climate_input_dim": 6,
  "climate_seq_len": 5,
  "dem": false,
  "feature_dim": 512,
  "hidden_channels": 64,
  "in_channels": 6,
  "kernel_size": 3,
  "lr": 5e-06,
  "lstm_hidden_dim": 128,
  "model_type": "actu",
  "n_classes": 3,
  "num_lstm_layers": 1,
  "original_resolution": [
    256,
    256
  ],
  "padding": 1,
  "skip_connections": true,
  "stride": 1,
  "target_length": 1,
  "task": "segmentation",
  "torch_dtype": "float32",
  "transformers_version": "4.53.2",
  "use_climate_branch": false,
  "use_dem_input": false,
  "auto_map": {
    "AutoModel": "modeling_actu.ACTUForImageSegmentation",
    "AutoConfig": "modeling_actu.ACTUConfig"
  }
}