Commit
·
5efb9ac
1
Parent(s):
5ec5a55
End of training
Browse files- README.md +78 -1
- all_results.json +85 -0
- eval_results.json +80 -0
- train_results.json +8 -0
- trainer_state.json +40 -0
README.md
CHANGED
|
@@ -2,6 +2,8 @@
|
|
| 2 |
license: other
|
| 3 |
base_model: nvidia/mit-b0
|
| 4 |
tags:
|
|
|
|
|
|
|
| 5 |
- generated_from_trainer
|
| 6 |
model-index:
|
| 7 |
- name: Segments
|
|
@@ -13,7 +15,82 @@ should probably proofread and complete it, then remove this comment. -->
|
|
| 13 |
|
| 14 |
# Segments
|
| 15 |
|
| 16 |
-
This model is a fine-tuned version of [nvidia/mit-b0](https://huggingface.co/nvidia/mit-b0) on the
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 17 |
|
| 18 |
## Model description
|
| 19 |
|
|
|
|
| 2 |
license: other
|
| 3 |
base_model: nvidia/mit-b0
|
| 4 |
tags:
|
| 5 |
+
- image-segmentation
|
| 6 |
+
- vision
|
| 7 |
- generated_from_trainer
|
| 8 |
model-index:
|
| 9 |
- name: Segments
|
|
|
|
| 15 |
|
| 16 |
# Segments
|
| 17 |
|
| 18 |
+
This model is a fine-tuned version of [nvidia/mit-b0](https://huggingface.co/nvidia/mit-b0) on the segments/sidewalk-semantic dataset.
|
| 19 |
+
It achieves the following results on the evaluation set:
|
| 20 |
+
- Loss: 0.9830
|
| 21 |
+
- Mean Iou: 0.1931
|
| 22 |
+
- Mean Accuracy: 0.2401
|
| 23 |
+
- Overall Accuracy: 0.7586
|
| 24 |
+
- Accuracy Unlabeled: nan
|
| 25 |
+
- Accuracy Flat-road: 0.7259
|
| 26 |
+
- Accuracy Flat-sidewalk: 0.9518
|
| 27 |
+
- Accuracy Flat-crosswalk: 0.5588
|
| 28 |
+
- Accuracy Flat-cyclinglane: 0.5550
|
| 29 |
+
- Accuracy Flat-parkingdriveway: 0.1159
|
| 30 |
+
- Accuracy Flat-railtrack: nan
|
| 31 |
+
- Accuracy Flat-curb: 0.1277
|
| 32 |
+
- Accuracy Human-person: 0.0990
|
| 33 |
+
- Accuracy Human-rider: 0.0
|
| 34 |
+
- Accuracy Vehicle-car: 0.9049
|
| 35 |
+
- Accuracy Vehicle-truck: 0.0
|
| 36 |
+
- Accuracy Vehicle-bus: 0.0
|
| 37 |
+
- Accuracy Vehicle-tramtrain: nan
|
| 38 |
+
- Accuracy Vehicle-motorcycle: 0.0
|
| 39 |
+
- Accuracy Vehicle-bicycle: 0.0
|
| 40 |
+
- Accuracy Vehicle-caravan: 0.0
|
| 41 |
+
- Accuracy Vehicle-cartrailer: 0.0
|
| 42 |
+
- Accuracy Construction-building: 0.8590
|
| 43 |
+
- Accuracy Construction-door: 0.0
|
| 44 |
+
- Accuracy Construction-wall: 0.0013
|
| 45 |
+
- Accuracy Construction-fenceguardrail: 0.0
|
| 46 |
+
- Accuracy Construction-bridge: 0.0
|
| 47 |
+
- Accuracy Construction-tunnel: nan
|
| 48 |
+
- Accuracy Construction-stairs: 0.0
|
| 49 |
+
- Accuracy Object-pole: 0.0088
|
| 50 |
+
- Accuracy Object-trafficsign: 0.0
|
| 51 |
+
- Accuracy Object-trafficlight: 0.0
|
| 52 |
+
- Accuracy Nature-vegetation: 0.9206
|
| 53 |
+
- Accuracy Nature-terrain: 0.7756
|
| 54 |
+
- Accuracy Sky: 0.8391
|
| 55 |
+
- Accuracy Void-ground: 0.0
|
| 56 |
+
- Accuracy Void-dynamic: 0.0
|
| 57 |
+
- Accuracy Void-static: 0.0
|
| 58 |
+
- Accuracy Void-unclear: 0.0
|
| 59 |
+
- Iou Unlabeled: nan
|
| 60 |
+
- Iou Flat-road: 0.5951
|
| 61 |
+
- Iou Flat-sidewalk: 0.7822
|
| 62 |
+
- Iou Flat-crosswalk: 0.5498
|
| 63 |
+
- Iou Flat-cyclinglane: 0.4666
|
| 64 |
+
- Iou Flat-parkingdriveway: 0.1001
|
| 65 |
+
- Iou Flat-railtrack: nan
|
| 66 |
+
- Iou Flat-curb: 0.1078
|
| 67 |
+
- Iou Human-person: 0.0979
|
| 68 |
+
- Iou Human-rider: 0.0
|
| 69 |
+
- Iou Vehicle-car: 0.6265
|
| 70 |
+
- Iou Vehicle-truck: 0.0
|
| 71 |
+
- Iou Vehicle-bus: 0.0
|
| 72 |
+
- Iou Vehicle-tramtrain: nan
|
| 73 |
+
- Iou Vehicle-motorcycle: 0.0
|
| 74 |
+
- Iou Vehicle-bicycle: 0.0
|
| 75 |
+
- Iou Vehicle-caravan: 0.0
|
| 76 |
+
- Iou Vehicle-cartrailer: 0.0
|
| 77 |
+
- Iou Construction-building: 0.4997
|
| 78 |
+
- Iou Construction-door: 0.0
|
| 79 |
+
- Iou Construction-wall: 0.0013
|
| 80 |
+
- Iou Construction-fenceguardrail: 0.0
|
| 81 |
+
- Iou Construction-bridge: 0.0
|
| 82 |
+
- Iou Construction-tunnel: nan
|
| 83 |
+
- Iou Construction-stairs: 0.0
|
| 84 |
+
- Iou Object-pole: 0.0088
|
| 85 |
+
- Iou Object-trafficsign: 0.0
|
| 86 |
+
- Iou Object-trafficlight: 0.0
|
| 87 |
+
- Iou Nature-vegetation: 0.7405
|
| 88 |
+
- Iou Nature-terrain: 0.6034
|
| 89 |
+
- Iou Sky: 0.8052
|
| 90 |
+
- Iou Void-ground: 0.0
|
| 91 |
+
- Iou Void-dynamic: 0.0
|
| 92 |
+
- Iou Void-static: 0.0
|
| 93 |
+
- Iou Void-unclear: 0.0
|
| 94 |
|
| 95 |
## Model description
|
| 96 |
|
all_results.json
ADDED
|
@@ -0,0 +1,85 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"epoch": 19.81,
|
| 3 |
+
"eval_accuracy_construction-bridge": 0.0,
|
| 4 |
+
"eval_accuracy_construction-building": 0.8590146051251705,
|
| 5 |
+
"eval_accuracy_construction-door": 0.0,
|
| 6 |
+
"eval_accuracy_construction-fenceguardrail": 0.0,
|
| 7 |
+
"eval_accuracy_construction-stairs": 0.0,
|
| 8 |
+
"eval_accuracy_construction-tunnel": NaN,
|
| 9 |
+
"eval_accuracy_construction-wall": 0.001317080924782568,
|
| 10 |
+
"eval_accuracy_flat-crosswalk": 0.5587614814704475,
|
| 11 |
+
"eval_accuracy_flat-curb": 0.1276987072001905,
|
| 12 |
+
"eval_accuracy_flat-cyclinglane": 0.5549554426151009,
|
| 13 |
+
"eval_accuracy_flat-parkingdriveway": 0.11587664148520291,
|
| 14 |
+
"eval_accuracy_flat-railtrack": NaN,
|
| 15 |
+
"eval_accuracy_flat-road": 0.7259208131761419,
|
| 16 |
+
"eval_accuracy_flat-sidewalk": 0.9517671543102283,
|
| 17 |
+
"eval_accuracy_human-person": 0.09899381037890112,
|
| 18 |
+
"eval_accuracy_human-rider": 0.0,
|
| 19 |
+
"eval_accuracy_nature-terrain": 0.7756278458254022,
|
| 20 |
+
"eval_accuracy_nature-vegetation": 0.9205963171170912,
|
| 21 |
+
"eval_accuracy_object-pole": 0.008821574806525607,
|
| 22 |
+
"eval_accuracy_object-trafficlight": 0.0,
|
| 23 |
+
"eval_accuracy_object-trafficsign": 0.0,
|
| 24 |
+
"eval_accuracy_sky": 0.8390657685487607,
|
| 25 |
+
"eval_accuracy_unlabeled": NaN,
|
| 26 |
+
"eval_accuracy_vehicle-bicycle": 0.0,
|
| 27 |
+
"eval_accuracy_vehicle-bus": 0.0,
|
| 28 |
+
"eval_accuracy_vehicle-car": 0.9049240346951853,
|
| 29 |
+
"eval_accuracy_vehicle-caravan": 0.0,
|
| 30 |
+
"eval_accuracy_vehicle-cartrailer": 0.0,
|
| 31 |
+
"eval_accuracy_vehicle-motorcycle": 0.0,
|
| 32 |
+
"eval_accuracy_vehicle-tramtrain": NaN,
|
| 33 |
+
"eval_accuracy_vehicle-truck": 0.0,
|
| 34 |
+
"eval_accuracy_void-dynamic": 0.0,
|
| 35 |
+
"eval_accuracy_void-ground": 0.0,
|
| 36 |
+
"eval_accuracy_void-static": 0.0,
|
| 37 |
+
"eval_accuracy_void-unclear": 0.0,
|
| 38 |
+
"eval_iou_construction-bridge": 0.0,
|
| 39 |
+
"eval_iou_construction-building": 0.49966273102422204,
|
| 40 |
+
"eval_iou_construction-door": 0.0,
|
| 41 |
+
"eval_iou_construction-fenceguardrail": 0.0,
|
| 42 |
+
"eval_iou_construction-stairs": 0.0,
|
| 43 |
+
"eval_iou_construction-tunnel": NaN,
|
| 44 |
+
"eval_iou_construction-wall": 0.0013105616961982412,
|
| 45 |
+
"eval_iou_flat-crosswalk": 0.5497595515292523,
|
| 46 |
+
"eval_iou_flat-curb": 0.107839451025818,
|
| 47 |
+
"eval_iou_flat-cyclinglane": 0.4665512602202192,
|
| 48 |
+
"eval_iou_flat-parkingdriveway": 0.10007739238668793,
|
| 49 |
+
"eval_iou_flat-railtrack": NaN,
|
| 50 |
+
"eval_iou_flat-road": 0.5950889830014944,
|
| 51 |
+
"eval_iou_flat-sidewalk": 0.7821736374605205,
|
| 52 |
+
"eval_iou_human-person": 0.09788031780354503,
|
| 53 |
+
"eval_iou_human-rider": 0.0,
|
| 54 |
+
"eval_iou_nature-terrain": 0.603375343788221,
|
| 55 |
+
"eval_iou_nature-vegetation": 0.7404922773118722,
|
| 56 |
+
"eval_iou_object-pole": 0.008814276343092762,
|
| 57 |
+
"eval_iou_object-trafficlight": 0.0,
|
| 58 |
+
"eval_iou_object-trafficsign": 0.0,
|
| 59 |
+
"eval_iou_sky": 0.805219699788265,
|
| 60 |
+
"eval_iou_unlabeled": NaN,
|
| 61 |
+
"eval_iou_vehicle-bicycle": 0.0,
|
| 62 |
+
"eval_iou_vehicle-bus": 0.0,
|
| 63 |
+
"eval_iou_vehicle-car": 0.6265363619850278,
|
| 64 |
+
"eval_iou_vehicle-caravan": 0.0,
|
| 65 |
+
"eval_iou_vehicle-cartrailer": 0.0,
|
| 66 |
+
"eval_iou_vehicle-motorcycle": 0.0,
|
| 67 |
+
"eval_iou_vehicle-tramtrain": NaN,
|
| 68 |
+
"eval_iou_vehicle-truck": 0.0,
|
| 69 |
+
"eval_iou_void-dynamic": 0.0,
|
| 70 |
+
"eval_iou_void-ground": 0.0,
|
| 71 |
+
"eval_iou_void-static": 0.0,
|
| 72 |
+
"eval_iou_void-unclear": 0.0,
|
| 73 |
+
"eval_loss": 0.9830149412155151,
|
| 74 |
+
"eval_mean_accuracy": 0.24010778315093972,
|
| 75 |
+
"eval_mean_iou": 0.19305747888272376,
|
| 76 |
+
"eval_overall_accuracy": 0.7586429852069163,
|
| 77 |
+
"eval_runtime": 153.8717,
|
| 78 |
+
"eval_samples_per_second": 0.975,
|
| 79 |
+
"eval_steps_per_second": 0.123,
|
| 80 |
+
"total_flos": 2.9595008832425165e+17,
|
| 81 |
+
"train_loss": 1.403595557302799,
|
| 82 |
+
"train_runtime": 1785.2215,
|
| 83 |
+
"train_samples_per_second": 9.523,
|
| 84 |
+
"train_steps_per_second": 0.594
|
| 85 |
+
}
|
eval_results.json
ADDED
|
@@ -0,0 +1,80 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"epoch": 19.81,
|
| 3 |
+
"eval_accuracy_construction-bridge": 0.0,
|
| 4 |
+
"eval_accuracy_construction-building": 0.8590146051251705,
|
| 5 |
+
"eval_accuracy_construction-door": 0.0,
|
| 6 |
+
"eval_accuracy_construction-fenceguardrail": 0.0,
|
| 7 |
+
"eval_accuracy_construction-stairs": 0.0,
|
| 8 |
+
"eval_accuracy_construction-tunnel": NaN,
|
| 9 |
+
"eval_accuracy_construction-wall": 0.001317080924782568,
|
| 10 |
+
"eval_accuracy_flat-crosswalk": 0.5587614814704475,
|
| 11 |
+
"eval_accuracy_flat-curb": 0.1276987072001905,
|
| 12 |
+
"eval_accuracy_flat-cyclinglane": 0.5549554426151009,
|
| 13 |
+
"eval_accuracy_flat-parkingdriveway": 0.11587664148520291,
|
| 14 |
+
"eval_accuracy_flat-railtrack": NaN,
|
| 15 |
+
"eval_accuracy_flat-road": 0.7259208131761419,
|
| 16 |
+
"eval_accuracy_flat-sidewalk": 0.9517671543102283,
|
| 17 |
+
"eval_accuracy_human-person": 0.09899381037890112,
|
| 18 |
+
"eval_accuracy_human-rider": 0.0,
|
| 19 |
+
"eval_accuracy_nature-terrain": 0.7756278458254022,
|
| 20 |
+
"eval_accuracy_nature-vegetation": 0.9205963171170912,
|
| 21 |
+
"eval_accuracy_object-pole": 0.008821574806525607,
|
| 22 |
+
"eval_accuracy_object-trafficlight": 0.0,
|
| 23 |
+
"eval_accuracy_object-trafficsign": 0.0,
|
| 24 |
+
"eval_accuracy_sky": 0.8390657685487607,
|
| 25 |
+
"eval_accuracy_unlabeled": NaN,
|
| 26 |
+
"eval_accuracy_vehicle-bicycle": 0.0,
|
| 27 |
+
"eval_accuracy_vehicle-bus": 0.0,
|
| 28 |
+
"eval_accuracy_vehicle-car": 0.9049240346951853,
|
| 29 |
+
"eval_accuracy_vehicle-caravan": 0.0,
|
| 30 |
+
"eval_accuracy_vehicle-cartrailer": 0.0,
|
| 31 |
+
"eval_accuracy_vehicle-motorcycle": 0.0,
|
| 32 |
+
"eval_accuracy_vehicle-tramtrain": NaN,
|
| 33 |
+
"eval_accuracy_vehicle-truck": 0.0,
|
| 34 |
+
"eval_accuracy_void-dynamic": 0.0,
|
| 35 |
+
"eval_accuracy_void-ground": 0.0,
|
| 36 |
+
"eval_accuracy_void-static": 0.0,
|
| 37 |
+
"eval_accuracy_void-unclear": 0.0,
|
| 38 |
+
"eval_iou_construction-bridge": 0.0,
|
| 39 |
+
"eval_iou_construction-building": 0.49966273102422204,
|
| 40 |
+
"eval_iou_construction-door": 0.0,
|
| 41 |
+
"eval_iou_construction-fenceguardrail": 0.0,
|
| 42 |
+
"eval_iou_construction-stairs": 0.0,
|
| 43 |
+
"eval_iou_construction-tunnel": NaN,
|
| 44 |
+
"eval_iou_construction-wall": 0.0013105616961982412,
|
| 45 |
+
"eval_iou_flat-crosswalk": 0.5497595515292523,
|
| 46 |
+
"eval_iou_flat-curb": 0.107839451025818,
|
| 47 |
+
"eval_iou_flat-cyclinglane": 0.4665512602202192,
|
| 48 |
+
"eval_iou_flat-parkingdriveway": 0.10007739238668793,
|
| 49 |
+
"eval_iou_flat-railtrack": NaN,
|
| 50 |
+
"eval_iou_flat-road": 0.5950889830014944,
|
| 51 |
+
"eval_iou_flat-sidewalk": 0.7821736374605205,
|
| 52 |
+
"eval_iou_human-person": 0.09788031780354503,
|
| 53 |
+
"eval_iou_human-rider": 0.0,
|
| 54 |
+
"eval_iou_nature-terrain": 0.603375343788221,
|
| 55 |
+
"eval_iou_nature-vegetation": 0.7404922773118722,
|
| 56 |
+
"eval_iou_object-pole": 0.008814276343092762,
|
| 57 |
+
"eval_iou_object-trafficlight": 0.0,
|
| 58 |
+
"eval_iou_object-trafficsign": 0.0,
|
| 59 |
+
"eval_iou_sky": 0.805219699788265,
|
| 60 |
+
"eval_iou_unlabeled": NaN,
|
| 61 |
+
"eval_iou_vehicle-bicycle": 0.0,
|
| 62 |
+
"eval_iou_vehicle-bus": 0.0,
|
| 63 |
+
"eval_iou_vehicle-car": 0.6265363619850278,
|
| 64 |
+
"eval_iou_vehicle-caravan": 0.0,
|
| 65 |
+
"eval_iou_vehicle-cartrailer": 0.0,
|
| 66 |
+
"eval_iou_vehicle-motorcycle": 0.0,
|
| 67 |
+
"eval_iou_vehicle-tramtrain": NaN,
|
| 68 |
+
"eval_iou_vehicle-truck": 0.0,
|
| 69 |
+
"eval_iou_void-dynamic": 0.0,
|
| 70 |
+
"eval_iou_void-ground": 0.0,
|
| 71 |
+
"eval_iou_void-static": 0.0,
|
| 72 |
+
"eval_iou_void-unclear": 0.0,
|
| 73 |
+
"eval_loss": 0.9830149412155151,
|
| 74 |
+
"eval_mean_accuracy": 0.24010778315093972,
|
| 75 |
+
"eval_mean_iou": 0.19305747888272376,
|
| 76 |
+
"eval_overall_accuracy": 0.7586429852069163,
|
| 77 |
+
"eval_runtime": 153.8717,
|
| 78 |
+
"eval_samples_per_second": 0.975,
|
| 79 |
+
"eval_steps_per_second": 0.123
|
| 80 |
+
}
|
train_results.json
ADDED
|
@@ -0,0 +1,8 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"epoch": 19.81,
|
| 3 |
+
"total_flos": 2.9595008832425165e+17,
|
| 4 |
+
"train_loss": 1.403595557302799,
|
| 5 |
+
"train_runtime": 1785.2215,
|
| 6 |
+
"train_samples_per_second": 9.523,
|
| 7 |
+
"train_steps_per_second": 0.594
|
| 8 |
+
}
|
trainer_state.json
ADDED
|
@@ -0,0 +1,40 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"best_metric": null,
|
| 3 |
+
"best_model_checkpoint": null,
|
| 4 |
+
"epoch": 19.813084112149532,
|
| 5 |
+
"eval_steps": 500,
|
| 6 |
+
"global_step": 1060,
|
| 7 |
+
"is_hyper_param_search": false,
|
| 8 |
+
"is_local_process_zero": true,
|
| 9 |
+
"is_world_process_zero": true,
|
| 10 |
+
"log_history": [
|
| 11 |
+
{
|
| 12 |
+
"epoch": 9.35,
|
| 13 |
+
"learning_rate": 2.641509433962264e-05,
|
| 14 |
+
"loss": 1.7773,
|
| 15 |
+
"step": 500
|
| 16 |
+
},
|
| 17 |
+
{
|
| 18 |
+
"epoch": 18.69,
|
| 19 |
+
"learning_rate": 2.830188679245283e-06,
|
| 20 |
+
"loss": 1.082,
|
| 21 |
+
"step": 1000
|
| 22 |
+
},
|
| 23 |
+
{
|
| 24 |
+
"epoch": 19.81,
|
| 25 |
+
"step": 1060,
|
| 26 |
+
"total_flos": 2.9595008832425165e+17,
|
| 27 |
+
"train_loss": 1.403595557302799,
|
| 28 |
+
"train_runtime": 1785.2215,
|
| 29 |
+
"train_samples_per_second": 9.523,
|
| 30 |
+
"train_steps_per_second": 0.594
|
| 31 |
+
}
|
| 32 |
+
],
|
| 33 |
+
"logging_steps": 500,
|
| 34 |
+
"max_steps": 1060,
|
| 35 |
+
"num_train_epochs": 20,
|
| 36 |
+
"save_steps": 200,
|
| 37 |
+
"total_flos": 2.9595008832425165e+17,
|
| 38 |
+
"trial_name": null,
|
| 39 |
+
"trial_params": null
|
| 40 |
+
}
|