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