pytorch模型可视化
pytorch中model.save()保存的pth模型在Netron中不能画出结构与土,因此转化成onnx模型再画
from torch.autograd import Variable
dummy_input = Variable(torch.randn(10, 3, 224, 224)).cuda()#randn(batch_size,channel,w,h)
torch.onnx.export(model, dummy_input, "model.onnx", export_params=True
原创
2021-01-24 17:28:45 ·
268 阅读 ·
0 评论