Shpionus opened a new issue #8769: URL: https://github.com/apache/tvm/issues/8769
I am working on [alexnet model](https://github.com/onnx/models/raw/master/vision/classification/alexnet) v1.4. When I'm trying to tune the model, application fails with `autotvm`, app raises following error: `Cannot find infer layout for task group_conv2d_nchw.x86` error. ``` def tune_kernels(tasks, measure_option, log_filename, tuner="gridsearch", early_stopping=None): msg = "tune kernels" LOG_insert(results_log, formatLOG, msg, logging.INFO) for i, task in enumerate(tasks): prefix = "[Task %2d/%2d] " % (i + 1, len(tasks)) # create tuner if tuner == "xgb" or tuner == "xgb-rank": tuner_obj = XGBTuner(task, loss_type="rank") elif tuner == "ga": tuner_obj = GATuner(task, pop_size=50) elif tuner == "random": tuner_obj = RandomTuner(task) elif tuner == "gridsearch": tuner_obj = GridSearchTuner(task) else: raise ValueError("Invalid tuner: " + tuner) # do tuning # n_trial = len(task.config_space) n_trial = 5 tuner_obj.tune( n_trial=n_trial, early_stopping=early_stopping, measure_option=measure_option, callbacks=[ autotvm.callback.progress_bar(n_trial, prefix=prefix), autotvm.callback.log_to_file(log_filename), ], ) ``` -- 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. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
