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