Hugging Face's logo

sam1120
/
safety-utcustom-terrain

Image Segmentation
Transformers
PyTorch
TensorBoard
segformer
vision
generated_from_trainer
Inference Endpoints
Model card Files Metrics Community
sam1120 commited on
Commit
f59f882
•
1 Parent(s): bea60b0

Training in progress, step 5120

Browse files
Files changed (10) hide show
  1. checkpoint-5120/config.json +102 -0
  2. checkpoint-5120/optimizer.pt +3 -0
  3. checkpoint-5120/pytorch_model.bin +3 -0
  4. checkpoint-5120/rng_state.pth +3 -0
  5. checkpoint-5120/scheduler.pt +3 -0
  6. checkpoint-5120/trainer_state.json +0 -0
  7. checkpoint-5120/training_args.bin +3 -0
  8. pytorch_model.bin +1 -1
  9. runs/Dec09_15-09-22_robolidar/events.out.tfevents.1702156168.robolidar.4047176.0 +3 -0
  10. training_args.bin +1 -1
checkpoint-5120/config.json ADDED
@@ -0,0 +1,102 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "nvidia/mit-b5",
3
+ "architectures": [
4
+ "SegformerForSemanticSegmentation"
5
+ ],
6
+ "attention_probs_dropout_prob": 0.0,
7
+ "classifier_dropout_prob": 0.1,
8
+ "decoder_hidden_size": 768,
9
+ "depths": [
10
+ 3,
11
+ 6,
12
+ 40,
13
+ 3
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
+ 64,
26
+ 128,
27
+ 320,
28
+ 512
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-5120/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:937c1dc4195d5e825e9ac39ac309966eb043a02362652f63ee2547ffad82c529
3
+ size 677523903
checkpoint-5120/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:89af25218452e26510ea2638cb2f671c6c8dab5f28c7a81869ca1ed8e970c97f
3
+ size 338830281
checkpoint-5120/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:645ae9b4b3cb906395786881351003ebd43ac897ecf7a55fb0b8a698bddbc398
3
+ size 14575
checkpoint-5120/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7d0b37f3ed8de391d92efd9ea617b480daf0e11505aefceded66ec0cc5e1fd1a
3
+ size 627
checkpoint-5120/trainer_state.json ADDED
The diff for this file is too large to render. See raw diff
 
checkpoint-5120/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c0b46b7c763380e7373efe0f2d0bb924ccdcfaf9b3579944fe14e0db2b1a7ca9
3
+ size 4219
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6f63f8c7afa64c7d33b2aa1bb0adab04f1727822cde838414214985ac65c80e7
3
  size 338830281
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:89af25218452e26510ea2638cb2f671c6c8dab5f28c7a81869ca1ed8e970c97f
3
  size 338830281
runs/Dec09_15-09-22_robolidar/events.out.tfevents.1702156168.robolidar.4047176.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dc9e06162528bef49b31abcfa41d8cb2d22056862bc14e967801aedf477f0ff5
3
+ size 26246
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:890edd60a7c33df244e1bde513c4d060e456d4721defbb7eb425aad8bfa1287f
3
  size 4219
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c0b46b7c763380e7373efe0f2d0bb924ccdcfaf9b3579944fe14e0db2b1a7ca9
3
  size 4219