kubeflow / trainer / 14732318482 / 1
51%
master: 51%

DEFAULT BRANCH: master
Ran
Files 28
Run time 0s
Badge
Embed ▾
README BADGES
x

If you need to use a raster PNG badge, change the '.svg' to '.png' in the link

Markdown

Textile

RDoc

HTML

Rst
coverage: 27.6% (+1.4%) from 26.207%
14732318482.1

push

github

web-flow 
KEP-2401: Complement torch plugin to support torchtune config mutation (#2587)

* chore(plugin): Add torchtune-related constants & update current torch plugin.

Signed-off-by: Electronic-Waste <2690692950@qq.com>

* chore(plugin): Add EnforceMLPolicy for torchtune.

Signed-off-by: Electronic-Waste <2690692950@qq.com>

* chore(plugin): Add UTs in torch plugin.

Signed-off-by: Electronic-Waste <2690692950@qq.com>

* fix(test): fix error in torch plugin UTs.

Signed-off-by: Electronic-Waste <2690692950@qq.com>

* chore(plugin): Choose recipe according to numNodes & numProcPerNode & Args.

Signed-off-by: Electronic-Waste <2690692950@qq.com>

* chore(sdk): Add PretrainedModel enum type.

Signed-off-by: Electronic-Waste <2690692950@qq.com>

* chore(plugin): Add torchtune config arg.

Signed-off-by: Electronic-Waste <2690692950@qq.com>

* chore(test): add UT for single-device full fine-tuning with torchtune.

Signed-off-by: Electronic-Waste <2690692950@qq.com>

* chore(test): Add test for multi-nodes full fine-tuning with torchtune.

Signed-off-by: Electronic-Waste <2690692950@qq.com>

* chore(test): Update torch validate UTs.

Signed-off-by: Electronic-Waste <2690692950@qq.com>

* fix(lint): fix lint error.

Signed-off-by: Electronic-Waste <2690692950@qq.com>

* fix(sdk): remove pretrained model enum type in sdk.

Signed-off-by: Electronic-Waste <2690692950@qq.com>

* fix(plugin): retrieve model name from runtimeRef.

Signed-off-by: Electronic-Waste <2690692950@qq.com>

* fix(lint): fix typo.

Signed-off-by: Electronic-Waste <2690692950@qq.com>

* fix(plugin): make some adjustments according to the review.

Signed-off-by: Electronic-Waste <2690692950@qq.com>

* fix(sdk): remove runtime in get_trainer_crd_from_builtin_trainer.

Signed-off-by: Electronic-Waste <2690692950@qq.com>

* fix(plugin): pass PET_ env variables in torch plugin for torchtune.

Signed-off-by: Electronic-Waste <2690692950@qq.com>

* fix(plugin): add env validation for torchtune.

Signed-... (continued)

735 of 2663 relevant lines covered (27.6%)

0.33 hits per line

Source Files on job 0 - 14732318482.1