Open pi-null-mezon opened 3 months ago
I have failed to convert SwinTransformerV2 onnx (it works properly, I have tested) model into rknn.
rknn-toolkit versions I have tried: 1.7.3, 1.7.5
My script:
... # CONFIG rknn = RKNN(verbose=True, verbose_file=os.path.join(target_path, 'conversion.log')) rknn.config(quantize_input_node=True, mean_values=mean, std_values=std, quantized_dtype='asymmetric_affine-u8', target_platform='rv1126', batch_size=100) # LOAD ret = rknn.load_onnx(model=onnx_model_file, input_size_list=input_size, inputs=inputs, outputs=outputs) #..............................SUCCESS if ret != 0: print('Load ONNX model failed!') exit(ret) # BUILD RKNN MODEL ret = rknn.build(do_quantization=True, dataset=os.path.join(target_path, 'samples.txt'), pre_compile=False) #...............................................FAIL ...
conversion.log [Uploading conversion.log…]()
I have failed to convert SwinTransformerV2 onnx (it works properly, I have tested) model into rknn.
rknn-toolkit versions I have tried: 1.7.3, 1.7.5
My script:
conversion.log [Uploading conversion.log…]()