“load”: 不是“torch::jit”的成员

 

严重性    代码    说明    项目    文件    行    禁止显示状态
错误    C2039    “load”: 不是“torch::jit”的成员    simnet    d:\soft\libtorch-win-shared-with-deps-latest\build_person\main2.cpp    59    
 

 

解决：

加上头文件：

#include // One-stop header.

 

#include 

	torch::DeviceType device_type;

	if (torch::cuda::is_available()) {
		device_type = torch::kCUDA;
	}
	else {
		device_type = torch::kCPU;
	}
	torch::Device device(device_type);


	// Deserialize the ScriptModule from a file using torch::jit::load().
	//反序列化pytorch模型，并加载到module变量
	std::shared_ptr module = torch::jit::load(argv[1], device);

	assert(module != nullptr);

torch::jit 没有成员load