AutoMR-pangu / README.md
Alexhf825's picture
Update README.md
979785a verified
---
datasets:
- Alexhf825/dataset-AutoMR-pangu
language:
- en
metrics:
- accuracy
base_model:
- FreedomIntelligence/openPangu-Embedded-7B
---
# AutoMR for Pangu
This project equips the Pangu model with the AutoMR reasoning framework, optimized for Huawei Ascend hardware.
## 🌳 Project Structure
```
.
β”œβ”€β”€ AMC.SH
β”œβ”€β”€ automr
β”‚Β  Β β”œβ”€β”€ config.py
β”‚Β  Β β”œβ”€β”€ dag.py
β”‚Β  Β β”œβ”€β”€ data_loader.py
β”‚Β  Β β”œβ”€β”€ evaluator.py
β”‚Β  Β β”œβ”€β”€ __init__.py
β”‚Β  Β β”œβ”€β”€ model.py
β”‚Β  Β β”œβ”€β”€ strategies.py
β”‚Β  Β β”œβ”€β”€ trainer.py
β”‚Β   └── utils.py
β”œβ”€β”€ checkpoints
β”‚Β   └── MATH
β”œβ”€β”€ embedder_server.sh
β”œβ”€β”€ generator_server.sh
β”œβ”€β”€ main.py
β”œβ”€β”€ math_train.sh
└── processed_data
Β  Β  β”œβ”€β”€ AMC
Β  Β  └── MATH
```
## πŸ”§ 1. Installation
### a. Clone the Project Repository
```bash
hf download Alexhf825/AutoMR-pangu --local-dir AutoMR-pangu
cd AutoMR-pangu
```
### b. Install Dependencies
```
```
### c. Download Datasets
This command will download the datasets and place them in the `./processed_data` directory, matching the project structure.
```bash
hf download Alexhf825/dataset-AutoMR-pangu --repo-type=dataset --local-dir=./
```
## πŸš€ 2. Start the Servers
This project requires two services running in an OpenAI-API style. Please run the following commands in **two separate terminal sessions**.
**Start the Embedder Server:**
```bash
bash embedder_server.sh
```
**Start the Generator Server:**
```bash
bash generator_server.sh
```
## πŸ“ˆ 3. Run Evaluation
A pre-trained checkpoint (`MATH`) is provided. You can directly evaluate the model on the `AMC` dataset using the following command:
```bash
bash AMC.sh
```
## πŸ‹οΈ 4. Run Training
You can also train the model from scratch on the `MATH` dataset by running:
```bash
bash math_train.sh
```