leandron opened a new pull request #6537: URL: https://github.com/apache/incubator-tvm/pull/6537
Introduce a sub-command to drive auto-tuning via command line, using AutoTVM, to produce the tuning records that can be used at compile time, or to further improve other tuning sessions. It follow good practices present on the auto-tuning tutorials. In case you want to try it out (supposing you have your tvm built and working with this PR), you can have a look on the example below: ``` $ wget https://github.com/onnx/models/raw/master/vision/classification/resnet/model/resnet50-v2-7.onnx $ python -m tvm.driver.tvmc --verbose tune --trials 1 --output log1.txt --target="llvm" resnet50-v2-7.onnx (...messages...) WARNING:root:Attribute spatial is ignored in relay.sym.batch_norm [Task 1/23] Current/Best: 0.00/ 0.00 GFLOPS | Progress: (1/1) | 1.32 s Done. [Task 2/23] Current/Best: 0.00/ 0.00 GFLOPS | Progress: (1/1) | 1.23 s Done. [Task 3/23] Current/Best: 0.00/ 0.00 GFLOPS | Progress: (1/1) | 1.07 s Done. ... ``` cc @comaniac @masahi @FrozenGene for reviews if possible ---------------------------------------------------------------- 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]
