Hugging Face's logo

sam1120
/
safety-utcustom-terrain-b0

Image Segmentation
Transformers
PyTorch
TensorBoard
segformer
vision
Generated from Trainer
Inference Endpoints
Model card Files Metrics Community
sam1120 commited on
Commit
8c26097
•
1 Parent(s): 7c25139

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_21-20-16_robolidar/events.out.tfevents.1716258023.robolidar.3008813.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:b5b52aad4704a86f87f29b0a38904cb7650ce877d6ccf8614f5c76b1de0c4940
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:fde14d1cd3c84446d28f9135b5afd7472147615f9b19c557b9a40d745db64e19
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:6ae3e621e780039b6372f57185ebc8826ff51d8891b4d0fda6ee4b4df223c8c5
3
- size 4155
 
 
 
 
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:84c230fc306e9c4c1263c82cdd73696e4dfa52e63292cd185e362a6b48d0b179
3
  size 14944077
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1658de93856364f7ec136c91114fa05cca28ad59a1c29256c869f831eaa24361
3
  size 14944077
runs/May20_21-20-16_robolidar/events.out.tfevents.1716258023.robolidar.3008813.0 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:5f85206009893bc5033d85cccaf10fd5c27d898ba8a07a64dece503e5ebe7825
3
- size 1430970
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c6b905b90b1b4e075ef9378346bbe0f9ea38c038e4047954b137a3d43a3eb63f
3
+ size 1447056