pytorch
Dw126
这个作者很懒,什么都没留下…
展开
-
用自定义c++运算符扩展TorchScript
Extending TorchScript with Custom C++ Operators的要点更新中…PyTorch 1.0的发布,为 PyTorch 引入了一个新的编程模型,称为 TorchScript。TorchScript 是 Python 的一个子集,它可以由TorchScript 的编译器进行解析、编译和优化。进一步地,编译后的TorchScript 模型可以序列化为磁盘上的文件格式,该文件格式的推理,可以用纯 c + + (以及 Python)加载并运行。TorchScript 支持原创 2020-11-11 18:42:38 · 942 阅读 · 0 评论 -
libtorch初体验
准备:win10+VS2017pytorch 1.1.0(cuda10.0)libtorch 1.1.0(cuda10.0)cmake >3.12解压libtorch放到自己喜欢的位置pytorch简单实现一个MLP:mlp.pyfrom torch import nnclass MLP(nn.Module): def __init__(self, in_dim, hid_dim1, hid_dim2, out_dim): super(MLP, self)原创 2020-10-20 17:13:54 · 183 阅读 · 0 评论