Autotuning resnet-101

Questions
#1

Hello.

I have a modified resnet-101 model saved in ONNX format, is there any guide on how to autotune specifically for this network?

#2

There’s currently no direct tutorial to process a onnx tuning, but you can get all what you want in different tutorials.

Check the tutorials/frontend/from_onnx.py, you’ll find how to load an onnx model and get the transformed relay graph:

mod, params = relay.frontend.from_onnx(onnx_model, shape_dict)

Then from the tutorials/auto_scheduler/tune_network_x86.py, you’ll find how to tune the relay graph by AutoScheduler.

Or from the tutorials/autotvm/tune_relay_x86.py, you’ll find how to tune the relay graph by AutoTVM.

#3

Alright, thank you very much!