2020年11月4日
pytorch转onnx
摘要： 一个输入： input_tensor = torch.randn([1, 3, 256, 512]) print ("Exporting to ONNX: ", onnx_save_name) torch_onnx_out = torch.onnx.export(model, input_tenso 阅读全文
posted @ 2020-11-04 14:01 皮卡皮卡妞 阅读(5663) 评论(0) 推荐(0) 编辑