Hugging Face's logo

sam1120
/
safety-utcustom-terrain-b0-optim

Image Segmentation
Transformers
PyTorch
TensorBoard
segformer
vision
Generated from Trainer
Inference Endpoints
Model card Files Metrics Community
sam1120 commited on
Commit
288560d
•
1 Parent(s): 5204eff

Model save

Browse files
Files changed (9) hide show
  1. checkpoint-5200/config.json +0 -102
  2. checkpoint-5200/optimizer.pt +0 -3
  3. checkpoint-5200/pytorch_model.bin +0 -3
  4. checkpoint-5200/rng_state.pth +0 -3
  5. checkpoint-5200/scheduler.pt +0 -3
  6. checkpoint-5200/trainer_state.json +0 -0
  7. checkpoint-5200/training_args.bin +0 -3
  8. pytorch_model.bin +1 -1
  9. runs/May20_23-57-43_robolidar/events.out.tfevents.1716267470.robolidar.3441566.0 +2 -2
checkpoint-5200/config.json DELETED
@@ -1,102 +0,0 @@
1
- {
2
- "_name_or_path": "nvidia/mit-b0",
3
- "architectures": [
4
- "SegformerForSemanticSegmentation"
5
- ],
6
- "attention_probs_dropout_prob": 0.0,
7
- "classifier_dropout_prob": 0.1,
8
- "decoder_hidden_size": 256,
9
- "depths": [
10
- 2,
11
- 2,
12
- 2,
13
- 2
14
- ],
15
- "downsampling_rates": [
16
- 1,
17
- 4,
18
- 8,
19
- 16
20
- ],
21
- "drop_path_rate": 0.1,
22
- "hidden_act": "gelu",
23
- "hidden_dropout_prob": 0.0,
24
- "hidden_sizes": [
25
- 32,
26
- 64,
27
- 160,
28
- 256
29
- ],
30
- "id2label": {
31
- "0": "unlabeled",
32
- "1": "NAT",
33
- "2": "concrete",
34
- "3": "grass",
35
- "4": "speedway bricks",
36
- "5": "steel",
37
- "6": "rough concrete",
38
- "7": "dark bricks",
39
- "8": "road",
40
- "9": "rough red sidewalk",
41
- "10": "tiles",
42
- "11": "red bricks",
43
- "12": "concrete tiles",
44
- "13": "REST"
45
- },
46
- "image_size": 224,
47
- "initializer_range": 0.02,
48
- "label2id": {
49
- "NAT": 1,
50
- "REST": 13,
51
- "concrete": 2,
52
- "concrete tiles": 12,
53
- "dark bricks": 7,
54
- "grass": 3,
55
- "red bricks": 11,
56
- "road": 8,
57
- "rough concrete": 6,
58
- "rough red sidewalk": 9,
59
- "speedway bricks": 4,
60
- "steel": 5,
61
- "tiles": 10,
62
- "unlabeled": 0
63
- },
64
- "layer_norm_eps": 1e-06,
65
- "mlp_ratios": [
66
- 4,
67
- 4,
68
- 4,
69
- 4
70
- ],
71
- "model_type": "segformer",
72
- "num_attention_heads": [
73
- 1,
74
- 2,
75
- 5,
76
- 8
77
- ],
78
- "num_channels": 3,
79
- "num_encoder_blocks": 4,
80
- "patch_sizes": [
81
- 7,
82
- 3,
83
- 3,
84
- 3
85
- ],
86
- "reshape_last_stage": true,
87
- "semantic_loss_ignore_index": 255,
88
- "sr_ratios": [
89
- 8,
90
- 4,
91
- 2,
92
- 1
93
- ],
94
- "strides": [
95
- 4,
96
- 2,
97
- 2,
98
- 2
99
- ],
100
- "torch_dtype": "float32",
101
- "transformers_version": "4.30.2"
102
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
checkpoint-5200/optimizer.pt DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:4f6c65248639839379b4f5b7301413d0d2d1723105ae6247a5508ee12f177926
3
- size 29861573
 
 
 
 
checkpoint-5200/pytorch_model.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:6d350ba477f0377dbb1c99e1eddbbf2bb5a86a268da1d685ce90f9d449c0ff0e
3
- size 14944077
 
 
 
 
checkpoint-5200/rng_state.pth DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:7824cd05d1990f9de976bb156bd34a7d1dea058e8051704fccd1f8c515c7520f
3
- size 14575
 
 
 
 
checkpoint-5200/scheduler.pt DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:0a682f639b8d011fdd980ce8988d51000a519eac2998c9a7eb5f22058ad08c79
3
- size 627
 
 
 
 
checkpoint-5200/trainer_state.json DELETED
The diff for this file is too large to render. See raw diff
 
checkpoint-5200/training_args.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:7935fa87beea0fca0caeec5cfb1e8512ed5be3a31fff1b806e2d5979c9b2318c
3
- size 4155
 
 
 
 
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:738040b0350f0999e598b39dac9cd347036141294ffd8a5538fea0e7acc02cc9
3
  size 14944077
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:912c08873d7e821cb5df046683c6b75c02833935e29dc9f3d9b4560a3e72b61f
3
  size 14944077
runs/May20_23-57-43_robolidar/events.out.tfevents.1716267470.robolidar.3441566.0 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:20a75756dbbe6bd49aac0e21dcd6ee02efac0ef9a02e0e5d9f1b5a0ce4678942
3
- size 1430994
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b467a3a326a08b04bc84fcd7f2b188cbe99dd1685659bbc90e1d4aa0347007f6
3
+ size 1447080