pytorch技巧

pytorch JIT浅解析

概要 Torch Script中的核心数据结构是ScriptModule 它是Torch的nn Module的类似物代表整个模型作为子模块树与普通模块一样 ScriptModule中的每个单独模块都可以包含子模块参数和方法在nn M

pytorch技巧 Pytorch JIT