Hi,
The documentation at tune_relay_arm mentions the use of use_spatial_pack_depthwise
. However, the tune_tasks
function doesn’t have such an option/code in the page mentioned.
I did come across the following function which has been mentioned in very old dated posts which mention the following code :
if spatial_pack:
for i in range(len(tasks)):
if tasks[i].name == 'topi_nn_depthwise_conv2d_nchw':
tsk = autotvm.task.create(tasks[i].name, tasks[i].args,
tasks[i].target, tasks[i].target_host,
'contrib_spatial_pack')
tasks[i] = tsk
Is the tutorial not up to date or has the use of use_spatial_pack_depthwise been deprecated. Either way could someone please point out the best way to autotune a mobilenet architecture for a Raspberry Pi 3B +. Any help would be much appreciated.
Thanks