如何在LibTorch中从IValue提取torch::nn::Sequential类型属性？

你遇到的问题核心是JIT加载的模型属性属于torch::jit::script::Module类型，无法直接转换为C++原生的torch::nn::Sequential——后者是LibTorch的C++ API模块，和JIT脚本模块不属于同一体系，所以to<torch::nn::Sequential>()会触发编译错误。

正确的处理方式分两种场景：

直接操作JIT脚本模块（推荐） #

不需要转换成C++原生Sequential，直接对JIT模块进行修改：

1. 先获取features对应的JIT模块：

   auto features_jit = mobilenetv2.attr("features").to<torch::jit::script::Module>();
   

2. 访问并修改子层，比如修改第一个卷积层的参数：

   // 这里的"0"对应Python中Sequential的第一个元素索引
   auto conv1 = features_jit.get_submodule("0").to<torch::nn::Conv2dImpl>();
   // 修改权重，示例：将权重置为1
   conv1->weight.data().fill_(1.0);
   

3. 如果需要遍历所有子模块：

   for (const auto& named_module : features_jit.named_modules()) {
       const std::string& name = named_module.name;
       torch::jit::script::Module sub_module = named_module.value;
       // 根据模块类型做处理，比如判断是否是Conv2d
       if (sub_module.is_instance<torch::nn::Conv2dImpl>()) {
           auto conv = sub_module.to<torch::nn::Conv2dImpl>();
           // 执行修改操作
       }
   }
   

转换为C++原生Sequential（可选） #

如果一定要用LibTorch的原生Sequential，需要手动遍历JIT模块的子元素并构建：

torch::nn::Sequential native_seq;
// 遍历JIT Sequential的所有子模块
for (size_t i = 0; ; ++i) {
    try {
        std::string idx_str = std::to_string(i);
        auto sub_jit = features_jit.get_submodule(idx_str);
        // 将JIT模块转换为ModuleHolder后加入原生Sequential
        native_seq->push_back(torch::nn::ModuleHolder<torch::nn::Module>(sub_jit.to<torch::nn::Module>()));
    } catch (const c10::Error& e) {
        // 捕获索引越界错误，停止遍历
        break;
    }
}

注意：转换为原生Sequential后，原JIT模型的修改不会同步到这个原生模块，反之亦然，所以除非有特殊需求，优先直接操作JIT模块。

内容的提问来源于stack exchange，提问作者Matt Eding