mirror of
https://github.com/wassname/Open-Assistant.git
synced 2026-09-13 12:11:30 +08:00
[fix] pre-commit update
This commit is contained in:
@@ -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
|
||||
|
||||
|
||||
Reference in New Issue
Block a user