Failed to execute script pegasus

[Electr1]

@Agent_kapo same issue discussed here, please check the thread to obtain the solution

Khadas VIM 3 NPU issues

Hello @Arjun_Gupta , [image] First, strides have three, [8, 16, 32]. [image] Second, the predictions you reshape into [144, 8400], so the len(predictions) is 144 and the shape of predictions[i][80:] is[1, 8320]. Third, you need to do softmax for four part separately. Divide [64, :] into [16, :]×4 and do softmax for each part. Last, you missed this step. Blockquote Then, each part does softmax and multiplies a constant matrix. The constant matrix is 0-15, shape(1, 16). The constant m…