Hugging Face's logo

blaz-r
/
BTC-B_clcd

Safetensors
model_hub_mixin
pytorch_model_hub_mixin
Model card Files
xet
 Community
BTC-B_clcd
File size: 2,732 Bytes
37738b9
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
{
  "config": {
    "ckpt_path": null,
    "config_tag": "BTC-B",
    "data": {
      "batch_size": 32,
      "data_path": "./datasets",
      "dataset": "clcd",
      "img_size": 256,
      "load_in_mem": null,
      "num_workers": 8,
      "pin_memory": true,
      "use_hf": true
    },
    "decoder": {
      "train": {
        "class_path": "models.modules.upernet.UperNetHead",
        "init_args": {
          "hidden_size": 512,
          "load_weights": null,
          "lr": null,
          "out_channels": 1,
          "transfer": true,
          "use_auxfcn": false,
          "weight_decay": null
        }
      }
    },
    "dev": false,
    "devices": "auto",
    "diff": {
      "class_path": "models.modules.simple_diff.SubDiff",
      "init_args": {
        "lr": null,
        "norm": false,
        "transfer": true,
        "use_in_pretrain": false,
        "weight_decay": null
      }
    },
    "encoder": {
      "class_path": "models.modules.HFBackbone",
      "init_args": {
        "lr": null,
        "lr_adjustments": false,
        "model_name": "facebook/mask2former-swin-base-IN21k-cityscapes-semantic",
        "pretrain_args": null,
        "return_layers": [
          1,
          2,
          3,
          4
        ],
        "train_args": {
          "finetune": true,
          "pretrained": true
        },
        "transfer": true,
        "weight_decay": null
      }
    },
    "eval_only": true,
    "out_proc": null,
    "pre_diff": null,
    "res_path": "./results",
    "seed": 42,
    "tag": "",
    "train": {
      "base_lr": 0.0001,
      "ckpt": null,
      "epochs": 100,
      "grad_clip_val": null,
      "layer_decay": 0.99,
      "loss": [
        "dice"
      ],
      "lr_scheduler": {
        "gamma": null,
        "ratio_t0": 1.0,
        "type": "cosine",
        "warmup_epoch_ratio": null,
        "warmup_lr_ratio": null
      },
      "transforms": [
        {
          "VerticalFlip": {
            "p": 0.3
          }
        },
        {
          "HorizontalFlip": {
            "p": 0.3
          }
        },
        {
          "Rotate": {
            "limit": [
              -90,
              90
            ],
            "p": 0.3
          }
        },
        {
          "Normalize": {
            "mean": [
              0.485,
              0.456,
              0.406
            ],
            "std": [
              0.229,
              0.224,
              0.225
            ]
          }
        }
      ],
      "val_freq": 5,
      "weight_decay": 0.0001
    },
    "vis_path": true,
    "wandb_proj": "None"
  }
}