Pytorch
Yew1168
这个作者很懒,什么都没留下…
展开
-
pytorch源码:C拓展
Python C拓展和普通的CPython方式拓展类似,下面主要看pytorch中的拓展模块“_C"的定义和相应其他模块的添加方式。pytorch中的拓展模块定义代码主要在torch/csrc/Module.cpp中,在预备知识中熟悉Python如何拓展C后,直接在Module.cpp找到感兴趣的初始化部分:#if PY_MAJOR_VERSION == 2PyMODINIT_FUNC...转载 2018-12-19 00:30:57 · 471 阅读 · 0 评论 -
RuntimeError: An attempt has been made to start a new process before the current process...
https://pytorch.org/tutorials/beginner/blitz/cifar10_tutorial.html#sphx-glr-beginner-blitz-cifar10-tutorial-pyimport torchimport torchvisionimport torchvision.transforms as transformsimport matp...原创 2018-12-19 01:21:19 · 20356 阅读 · 9 评论