This is an automated email from the ASF dual-hosted git repository.
tqchen pushed a change to branch unity
in repository https://gitbox.apache.org/repos/asf/tvm.git
from e47bf761d4 [Unity] Smart parameter fetching (#14708)
add d468419ea0 [Unity][Training] Trainer and SetupTrainer (#14706)
No new revisions were added by this update.
Summary of changes:
python/tvm/relax/training/__init__.py | 5 +-
python/tvm/relax/training/setup_trainer.py | 214 +++++++++++
python/tvm/relax/training/trainer.py | 393 +++++++++++++++++++++
tests/python/relax/test_training_setup_trainer.py | 229 ++++++++++++
.../python/relax/test_training_trainer_numeric.py | 170 +++++++++
5 files changed, 1010 insertions(+), 1 deletion(-)
create mode 100644 python/tvm/relax/training/setup_trainer.py
create mode 100644 python/tvm/relax/training/trainer.py
create mode 100644 tests/python/relax/test_training_setup_trainer.py
create mode 100644 tests/python/relax/test_training_trainer_numeric.py
