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)

目前不行。