DarthReca commited on
Commit
0289342
·
verified ·
1 Parent(s): e3f3842

Upload ACTUForImageSegmentation

Browse files
Files changed (2) hide show
  1. config.json +36 -0
  2. model.safetensors +3 -0
config.json ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "act_layer": "identity",
3
+ "architectures": [
4
+ "ACTUForImageSegmentation"
5
+ ],
6
+ "backbone": "convnextv2_base",
7
+ "batch_first": true,
8
+ "bias": true,
9
+ "bidirectional": false,
10
+ "climate": false,
11
+ "climate_input_dim": 6,
12
+ "climate_seq_len": 5,
13
+ "dem": false,
14
+ "feature_dim": 512,
15
+ "hidden_channels": 64,
16
+ "in_channels": 6,
17
+ "kernel_size": 3,
18
+ "lr": 5e-06,
19
+ "lstm_hidden_dim": 128,
20
+ "model_type": "actu",
21
+ "n_classes": 3,
22
+ "num_lstm_layers": 1,
23
+ "original_resolution": [
24
+ 256,
25
+ 256
26
+ ],
27
+ "padding": 1,
28
+ "skip_connections": true,
29
+ "stride": 1,
30
+ "target_length": 1,
31
+ "task": "segmentation",
32
+ "torch_dtype": "float32",
33
+ "transformers_version": "4.53.2",
34
+ "use_climate_branch": false,
35
+ "use_dem_input": false
36
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a3e210955dbff46e240bbdfbb39a7200021ce9774d6753194e6f117fac03ef46
3
+ size 889607564