Auto-scheduling Problem on CUDA

You can follow the scripts in this repo to correctly set the tuning parameters of Ansor

Note that Ansor works better for NHWC layout, so probably you can follow the scripts above to convert your model into NHWC layout.