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
336e349
•
1 Parent(s): b461ab9

Training in progress, step 3440

Browse files
Files changed (10) hide show
  1. checkpoint-3440/config.json +102 -0
  2. checkpoint-3440/optimizer.pt +3 -0
  3. checkpoint-3440/pytorch_model.bin +3 -0
  4. checkpoint-3440/rng_state.pth +3 -0
  5. checkpoint-3440/scheduler.pt +3 -0
  6. checkpoint-3440/trainer_state.json +0 -0
  7. checkpoint-3440/training_args.bin +3 -0
  8. pytorch_model.bin +1 -1
  9. runs/Dec09_11-39-12_robolidar/events.out.tfevents.1702143557.robolidar.3136372.0 +3 -0
  10. training_args.bin +1 -1
checkpoint-3440/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-3440/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4eaf5bea4560b270b065e57faebc2c641e8a224fdc0ca54264b9172c427f1c4a
3
+ size 677523903
checkpoint-3440/pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8ac9db328de18ce543b9e38afb6871a57d5c2bc3d2087207701871f0d1f10496
3
+ size 338830281
checkpoint-3440/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5c1ce7d63dadd7391b91d6ad3a1f1a72608e5a5296bca0770f5f8ddb890bced5
3
+ size 14575
checkpoint-3440/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c655404fcd644e2d7bb74a07537343a49b052892aa3f16b1c0abee1f94c31fb8
3
+ size 627
checkpoint-3440/trainer_state.json ADDED
The diff for this file is too large to render. See raw diff
 
checkpoint-3440/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:890edd60a7c33df244e1bde513c4d060e456d4721defbb7eb425aad8bfa1287f
3
+ size 4219
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4793446f5108bd132bc7590510622276dd0c697440142911f07bba169619f9ef
3
  size 338830281
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8ac9db328de18ce543b9e38afb6871a57d5c2bc3d2087207701871f0d1f10496
3
  size 338830281
runs/Dec09_11-39-12_robolidar/events.out.tfevents.1702143557.robolidar.3136372.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d078ad0f77d7868a93f04a34455deed0d33afc8e80854b3f148417b0f5e1c5c7
3
+ size 26246
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:33a912940e7e34efe4975f77b1cad679c8730d2fdf4b965d05dd2510fba4c6a1
3
  size 4219
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:890edd60a7c33df244e1bde513c4d060e456d4721defbb7eb425aad8bfa1287f
3
  size 4219