Upload ACTUForImageSegmentation

config.json

@@ -3,23 +3,19 @@
   "architectures": [
     "ACTUForImageSegmentation"
   ],
-  "auto_map": {
-    "AutoModel": "modeling_actu.ACTUForImageSegmentation",
-    "AutoConfig": "modeling_actu.ACTUConfig"
-  },
   "backbone": "convnextv2_base",
   "batch_first": true,
   "bias": true,
   "bidirectional": false,
-  "climate": true,
+  "climate": false,
   "climate_input_dim": 6,
   "climate_seq_len": 5,
-  "dem": true,
+  "dem": false,
   "feature_dim": 512,
   "hidden_channels": 64,
   "in_channels": 6,
   "kernel_size": 3,
-  "lr": 5e-06,
+  "lr": 0.0005,
   "lstm_hidden_dim": 128,
   "model_type": "actu",
   "n_classes": 1,
@@ -36,6 +32,6 @@
   "task": "segmentation",
   "torch_dtype": "float32",
   "transformers_version": "4.53.2",
-  "use_climate_branch": true,
-  "use_dem_input": true
+  "use_climate_branch": false,
+  "use_dem_input": false
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:755045f70cd0246ab63753c32973d106c30f656efa5025a88213ee36b1c8c394
-size 1163163820
+oid sha256:c9550b8467cab3f119f667f69577bb61fc42fc3d7fabbabc32de050f619b974b
+size 889598340