Hi all, I’m looking to run model training on VTA. However, I did not find any tutorials which talk about this. Do the tvm stack(and VTA) support training? If so, are there any example tests or guides to run the workload.
Thanks.
Yes, TVM and VTA do support model training, but it’s not as straightforward as inference. While most tutorials focus on inference, you can check out the TVM GitHub repo for example tests and guides. Look for tests/python/relay and tests/python/vta folders; they contain scripts that might help. Also, the TVM community forum is a great place to ask for specific examples and guidance.