Toybrick

标题: pytorch导出网络转化rknn问题 [打印本页]

作者: zwdtc 时间: 2020-7-31 10:20
标题: pytorch导出网络转化rknn问题
各位,在用pytorch生成的pt文件进而导出到rknn文件的流程中, pytorch网络导出只能用如下的torch.jit.trace吗?能用torch.save(net.state_dict())吗?
dummy_input = torch.randn([1, 3, 300, 300])
trace_model = torch.jit.trace(net, dummy_input)
PATH = './VOC0730.pt'
trace_model.save(PATH)

作者: leok 时间: 2020-8-3 09:03
目前不行。

欢迎光临 Toybrick (https://t.rock-chips.com/)