Can TVM compile and run train models?

Questions
1

Hi All! I start to study TVM weeks ago. I can’t find any train model in tutorials. Can TVM compile and run a train model like Bert, FasterRCNN or any other train model? Thank you so much~

2

We have done some tries on training a simple model like MLP by using the removed nnvm. It seems to work but there are many problems we meet. Next, we will use Relay to try training.

1 Like