pytorch网络转C++

pytorch JIT浅解析
1.用jit将python端网络模型转换到C++加载模型 用jit.trace比较简单
2.不会记录控制流(和参数相关的判断 if while…)
但是转换过程中也能正常转换,会按照当前输入,记录当前控制流的操作
eg:

	input = 1
	if input > 2:
		print("aaaaaaa")
	else:
		print("bbbbb")

此时只会记录print(“bbbbb”)的操作

3.网络中不能使用并行

# self.model = nn.DataParallel(self.model)

Torch 脚本_Github

C++ load pytorch 时的数据转换

PyTorch 1.0 中文官方教程:使用 PyTorch C++ 前端

PyTorch 1.0 中文官方教程:混合前端的 seq2seq 模型部署

你可能感兴趣的:(pytorch网络转C++)