vikp commited on
Commit
ca63704
·
verified ·
1 Parent(s): 507925c

Upload SegformerForRegressionMask

Browse files
Files changed (2) hide show
  1. config.json +3 -3
  2. model.safetensors +2 -2
config.json CHANGED
@@ -1,12 +1,12 @@
1
  {
2
- "_name_or_path": "line_detector_sd_arxiv/checkpoint-8000",
3
  "architectures": [
4
  "SegformerForRegressionMask"
5
  ],
6
  "attention_probs_dropout_prob": 0.0,
7
  "classifier_dropout_prob": 0.1,
8
- "decoder_hidden_size": 768,
9
- "decoder_layer_hidden_size": 192,
10
  "decoder_upsample_rate": 2,
11
  "depths": [
12
  3,
 
1
  {
2
+ "_name_or_path": "line_detector_smalldecoder2/checkpoint-44000",
3
  "architectures": [
4
  "SegformerForRegressionMask"
5
  ],
6
  "attention_probs_dropout_prob": 0.0,
7
  "classifier_dropout_prob": 0.1,
8
+ "decoder_hidden_size": 1024,
9
+ "decoder_layer_hidden_size": 256,
10
  "decoder_upsample_rate": 2,
11
  "depths": [
12
  3,
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e301d3f73bbda47cecbe02da764348830a917eb046885190aa964c1a3505ac09
3
- size 119881240
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:45a1bc4955da075aa75abecc6a420d43ed27a154a298c6a03caf96443937005a
3
+ size 121985624