Skip to content

Commit 5551da4

Browse files
committed
added a plot from training
1 parent b7f3da7 commit 5551da4

File tree

2 files changed

+7
-4
lines changed

2 files changed

+7
-4
lines changed

README.md

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -67,14 +67,17 @@ kernel-coder (GRPO-trained) 🚧 TBD 🚧 TBD
6767

6868
We test on label 1 (100 test cases) with temperature 1.0 and top_p 0.97. Preliminary results show the importance of Triton-specific training for compilation and correctness.
6969

70+
Here is the training progress plot at 100 steps:
71+
72+
![kernel-coder_1](assets/kernel-coder_1.png)
7073

7174
## 📂 Code Structure
7275

73-
Here are two main components of the code:
74-
1. `nano_r1_script.py` is originally from [nano-aha-moment](https://github.com/McGill-NLP/nano-aha-moment/blob/f6384878831796fc29f560016e3cd570d264b823/nano_r1_script.py) and modified for our project, `kernel-coder`
75-
2. [`KernelBench`](https://github.com/insop/KernelBench) is forked from [this repo](https://github.com/ScalingIntelligence/KernelBench) and modified for our project
76+
The codebase consists of two main components:
77+
1. `nano_r1_script.py` - Modified from [nano-aha-moment](https://github.com/McGill-NLP/nano-aha-moment/blob/f6384878831796fc29f560016e3cd570d264b823/nano_r1_script.py) for our `kernel-coder` project
78+
2. [`KernelBench`](https://github.com/insop/KernelBench) - Forked and modified from [ScalingIntelligence/KernelBench](https://github.com/ScalingIntelligence/KernelBench)
7679

77-
Here is the repository structure of the project.
80+
Project structure:
7881

7982
```
8083
kernel-coder/

assets/kernel-coder_1.png

426 KB
Loading

0 commit comments

Comments
 (0)