Upload root_scripts/read_reward.py with huggingface_hub
Browse files- root_scripts/read_reward.py +11 -0
root_scripts/read_reward.py
ADDED
|
@@ -0,0 +1,11 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
|
| 2 |
+
import verl, os
|
| 3 |
+
verl_dir = os.path.dirname(verl.__file__)
|
| 4 |
+
|
| 5 |
+
# Read reward.py
|
| 6 |
+
reward_file = os.path.join(verl_dir, "trainer", "ppo", "reward.py")
|
| 7 |
+
if os.path.exists(reward_file):
|
| 8 |
+
with open(reward_file) as f:
|
| 9 |
+
print(f.read()[:3000])
|
| 10 |
+
else:
|
| 11 |
+
print("reward.py not found")
|