- rknn = RKNN(verbose=True)
- rknn.config(batch_size=1,
- epochs=1) # asymmetric_quantized-u8, quantized_dtype='dynamic_fixed_point-16' , channel_mean_value='0 0 0 1', reorder_channel='0 1 2',
- system = platform.system()
- seq_len = 64
- isize = 672
- # hsize = 6
- # hsize1 = 4
- pt_file = 'toml_fp32_acoustic.pt'
- #pt_file = 'toml_fp32_feat.pt'
- m = torch.jit.load(pt_file)
- m.eval()
- #x = torch.rand((1, seq_len, isize))
- x = torch.linspace(-1, 1, steps=(isize * seq_len))
- x = x.view(1, seq_len, -1)
- input_size_list = [[1, seq_len, isize]]
- #x1 = torch.full((1, hsize), fill_value=0.2, dtype=torch.float)
- ix = x.numpy()
- out = m.forward(x)
- print("torch out:")
- print(out)
- rknn.load_pytorch(model=pt_file, input_size_list=input_size_list)
- rknn_file = pt_file + '.rknn'
- print('--> Building model')
- ret = rknn.build(do_quantization=False, dataset='./dataset.txt')
- if ret != 0:
- print('Build pytorch failed!')
- exit(ret)
- print('build done')
- rm(rknn_file)
- rknn.export_rknn(rknn_file)
复制代码
jefferyzhang 发表于 2020-3-12 17:05
pytorch版本是多少?
rknn那边只能支持1.2的
jefferyzhang 发表于 2020-3-12 17:19
问题已经上报给NPU部门了,等待回复
jefferyzhang 发表于 2020-3-12 17:19
问题已经上报给NPU部门了,等待回复
jefferyzhang 发表于 2020-3-12 17:19
问题已经上报给NPU部门了,等待回复
yjcn 发表于 2020-3-27 21:52
您好 请问这个问题还没有得到回复吗?
jefferyzhang 发表于 2020-5-8 08:31
NPU Team回复:
该问题已经解决,下一个版本发布就可以用了。
iamher0 发表于 2020-6-10 13:41
pytorch==1.2.0
rknn-toolkit==1.3.2
存在相同的问题
jefferyzhang 发表于 2020-6-10 14:31
下一个版本指的是1.3.2以后的版本。。。还没发布
jefferyzhang 发表于 2020-5-8 08:31
NPU Team回复:
该问题已经解决,下一个版本发布就可以用了。
午后阳光 发表于 2020-8-19 17:14
您好,我也遇到了同样的问题,至于您这边说的该问题已解决,下一个版本就可以用,这里的下一个版本指的是 ...
jefferyzhang 发表于 2020-8-20 08:35
1.3.2以后的版本。
置顶帖有1.4.0preview,你先试试
午后阳光 发表于 2020-8-20 16:10
/envs/toolkit140/lib/python3.6/site-packages/onnx_tf/common/__init__.py:87: UserWarning: FrontendHa ...
jefferyzhang 发表于 2020-5-8 08:31
NPU Team回复:
该问题已经解决,下一个版本发布就可以用了。
yang_allen 发表于 2020-8-20 16:57
大概什么时候发布下一个版本哟,更新rknn的版本吗
jefferyzhang 发表于 2020-5-8 08:31
NPU Team回复:
该问题已经解决,下一个版本发布就可以用了。
欢迎光临 Toybrick (https://t.rock-chips.com/) | Powered by Discuz! X3.3 |