File size: 1,884 Bytes
99cae4e
 
 
 
 
 
 
 
 
e8c1d9c
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
7a35214
 
e8c1d9c
 
 
 
 
 
979785a
e8c1d9c
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
---
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
```