merrymercy opened a new pull request #6270: URL: https://github.com/apache/incubator-tvm/pull/6270
For the full upstream plan, see [Ansor RFC](https://discuss.tvm.ai/t/rfc-ansor-an-auto-scheduler-for-tvm-autotvm-v2-0/7005/32). This PR adds a xgboost-based cost model. It is similar to the existing xgboost model in autotvm but works on the more general feature representation introduced by #6190 . RMSE is used as the loss function, but the general feature representation needs slightly modification to the loss function. This PR implements a custom loss function "pack-sum-rmse" to support this. ---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: [email protected]
