pytorch训练的pt模型转换为onnx(nn.DataParallel()、model、model.state_dict())
pt转onnx流程与常见问题pt转onnx流程pt转onnx流程1、读取pt模型文件,文件既可以是torch.save(model,path)整体保存的模型,也可以是保存的字典文件。//Anhighlightedblockdefload_model(model,model_path):checkpoint=torch.load(model_path,map_location=lambdastora