# 自己的模型为：model
traced_script_module = torch.jit.trace(model, src_inp)
traced_script_module.save("model.pt")
# 安装libtorch
# 使用pytorch训练模型时，如果模型里有BN或者Dropout层，测试之前需要加上 model = model.eval()
更完整的方法：https://zhuanlan.zhihu.com/p/191569603
# 自己的模型为：model
traced_script_module = torch.jit.trace(model, src_inp)
traced_script_module.save("model.pt")
# 安装libtorch
# 使用pytorch训练模型时，如果模型里有BN或者Dropout层，测试之前需要加上 model = model.eval()
更完整的方法：https://zhuanlan.zhihu.com/p/191569603
上一篇 sparksql和hive yyyy-mm-dd和yyyymmdd之间的转化问题
下一篇 【clickhouse】未解决 ClickHouse exception, code: 1002 DB::Exception: Directory already exists
名师互学网 版权所有 (c)2021-2022 ICP备案号：晋ICP备2021003244-6号
