[fix] pre-commit update

This commit is contained in:
theblackcat102
2023-01-01 11:43:15 +00:00
parent 1ddd9155f9
commit fe99b46f2e
15 changed files with 337 additions and 294 deletions
+4 -7
View File
@@ -2,7 +2,6 @@
Trainer code based on huggingface. Compatible with deepspeed or accelerate
Requirements
```
@@ -15,12 +14,10 @@ torch==1.12
Start training reward model
```bash
python trainer.py configs/electra-base-dis-webgpt.yml
```
Additional axis labeling, this outputs a 4 summary quality evaluation metrics (score are normalized to 0-1 )
```bash
@@ -29,13 +26,13 @@ python summary_quality_trainer.py configs/test-bloomz-560m-quality.yml
The four summary are :
* overall
- overall
* accuracy
- accuracy
* coverage
- coverage
* coherence
- coherence
## Dataset