Added the README and script files for training sql_agent on NPU #272
Conversation
|
/cla |
|
@microsoft-github-policy-service agree company="HUAWEI" |
| @@ -0,0 +1,200 @@ | |||
| """Train an SQL agent on the Spider dataset using Agent-lightning. | |||
There was a problem hiding this comment.
what's the difference of this new script and the old script?
There was a problem hiding this comment.
Added configuration for training on NPUs.
There was a problem hiding this comment.
Added "device": "npu" to the trainer and modified several other parameters.
There was a problem hiding this comment.
In that case, create a new function in like config_train_npu in the original script and add a section in the README will be enough. You can reference this README in the spider recipe.
There was a problem hiding this comment.
We don't want to include a dedicated script in a CI-maintained examples for some hardwares that is not included in the CI., unless we will have a NPU machine for daily testing.
2 participants
Added the README and script files for training sql_agent on NPU