Add kaggle/train_grpo.py
Browse files- kaggle/train_grpo.py +1 -1
kaggle/train_grpo.py
CHANGED
|
@@ -394,7 +394,6 @@ def run_heldout_agent_eval(
|
|
| 394 |
"resolution": "ERROR",
|
| 395 |
}
|
| 396 |
)
|
| 397 |
-
_vlog("heldout_eval.done", n=len(rows))
|
| 398 |
finally:
|
| 399 |
try:
|
| 400 |
env._end_trace()
|
|
@@ -404,6 +403,7 @@ def run_heldout_agent_eval(
|
|
| 404 |
env._cleanup()
|
| 405 |
except Exception:
|
| 406 |
pass
|
|
|
|
| 407 |
return rows
|
| 408 |
|
| 409 |
|
|
|
|
| 394 |
"resolution": "ERROR",
|
| 395 |
}
|
| 396 |
)
|
|
|
|
| 397 |
finally:
|
| 398 |
try:
|
| 399 |
env._end_trace()
|
|
|
|
| 403 |
env._cleanup()
|
| 404 |
except Exception:
|
| 405 |
pass
|
| 406 |
+
_vlog("heldout_eval.done", n=len(rows))
|
| 407 |
return rows
|
| 408 |
|
| 409 |
|