shrinath-suresh
|
c5325db7d2
|
feat(components): Adding Trainer component for PyTorch - KFP (#5767)
* Create README.md
Initial Readme
* Create README.md
Initial commit for the PyTorch pipeline examples
* Update README.md
* Adding PyTorch training component
Signed-off-by: Shrinath Suresh <shrinath@ideas2it.com>
* Adding PyTorch - trainer unit tests
Signed-off-by: Shrinath Suresh <shrinath@ideas2it.com>
* Fixing UT and adding setup.py
Signed-off-by: ankan94 <ankan@ideas2it.com>
* update setup.py, trainer compopents pyfiles,applied pylyintrc
Signed-off-by: Arvind-Ideas2IT <arvindkumarsingh.gautam@ideas2it.com>
* adressed the review comments,applied pylint, black
Signed-off-by: Arvind-Ideas2IT <arvindkumarsingh.gautam@ideas2it.com>
* added copyright headers in tests files
Signed-off-by: Arvind-Ideas2IT <arvindkumarsingh.gautam@ideas2it.com>
* updated base component
Signed-off-by: Arvind-Ideas2IT <arvindkumarsingh.gautam@ideas2it.com>
* Changing to Apache2 License
Reviewed license change with Legal team
* Switch to Apache2 License
* Add tox for test automation with presubmit script
* Add model archiver to dependencies
* Cleanup setup.py
* Cleanup
* Cleanup
* Cleanup
* Using common fixture for unit tests and adding gpu fix for saving the model
Signed-off-by: Shrinath Suresh <shrinath@ideas2it.com>
* Changing to Apache2 License
Signed-off-by: Arvind-Ideas2IT <arvindkumarsingh.gautam@ideas2it.com>
* resolve typo
Signed-off-by: Arvind-Ideas2IT <arvindkumarsingh.gautam@ideas2it.com>
* Added OWNERS
* OWENRS will be added as separate PR
* fix detect_version function in setup.py
Signed-off-by: Arvind-Ideas2IT <arvindkumarsingh.gautam@ideas2it.com>
* Cleanup for setup description field warnings
Co-authored-by: Geeta Chauhan <4461127+chauhang@users.noreply.github.com>
Co-authored-by: ankan94 <ankan@ideas2it.com>
Co-authored-by: Arvind-Ideas2IT <arvindkumarsingh.gautam@ideas2it.com>
Co-authored-by: Geeta Chauhan <gchauhan@fb.com>
|
2021-06-08 12:12:56 +08:00 |