pytorch
文章平均质量分 61
lshiwjx
哈?
展开
-
pytorch扩展
python__init__.py导入包时实际导入的是__init__.py,可以通过__init__.py来控制可以使用的包和函数。 __all__关键字可以方便得用来指定要导入的包。ctypescffi的wrapercffipytorchc文件需要一个h文件python文件build soffi.create_extension和build扩展autograd需要init forward 和原创 2017-09-24 16:47:20 · 274 阅读 · 0 评论 -
install for pytorch
tensorboard tensorflow-cpu pillow simd pandas原创 2017-09-18 15:03:49 · 480 阅读 · 0 评论 -
Pytorch常用函数
向量操作view(类似reshape):但是reshape always copies memory. view never copies memory网络层nn: 里面包含了大多数操作c扩展原创 2017-09-04 22:02:48 · 456 阅读 · 0 评论 -
Pytorch写深度学习
数据输入torch.utils.data.DataLoader:用来读取数据 ImageFolder:用来准备数据集 主要是getitem,init和len函数模型init和forwardGputorch.cuda.is_available() ~.cuda() 多GPU训练:DataParallel流程for data in data_loadervariableforwardlo原创 2017-09-04 22:18:23 · 568 阅读 · 0 评论 -
torch.nn
nn.Module模型的基类add_module(name, module) 等价于self.name = nn.xxx,模块使用名字来接触cpu() cuda() 复制到设备上double() float() half()eval() train() 只对BN和Dropout有用apply(fn) 应用于children的返回值,即所有子模型,可用于initstate_dict原创 2017-10-21 12:04:12 · 373 阅读 · 0 评论 -
Tensor
abs_() vs abs() : in_place 的区别原创 2017-10-21 12:10:34 · 177 阅读 · 0 评论 -
常见bug
THCUNN/ClassNLLCriterion.cu:57: void cunn_ClassNLLCriterion_updateOutput_kernel:class num 设置不对 RuntimeError: some of the strides of a given numpy array are negative. This is currently not supported, ...原创 2017-11-14 10:05:30 · 1231 阅读 · 0 评论 -
pytorch to caffe
安装caffe, gpu版本的,可以直接通过conda安装conda install caffe-gpupytorch安装:官网conda install pytorch torchvision cuda80 -c pytorchpytorch to caffe这里使用一个开源的代码: pytorchtocaffemodel_def = os.path.join(c...原创 2018-08-18 15:29:44 · 3956 阅读 · 6 评论 -
复制conda环境到另一台电脑,避免在线安装,并替换文件路径
复制conda环境到另一台电脑,可以避免在线安装直接复制envs中的内容即可,并修改~/.conda/environment.txt中的内容替换非binary文件中的文件路径find . -type f -exec grep -Iq . {} ; -exec sed -i ‘s/home/name1/home/name2/g’ {} ;其中-exec表示根据之前的结果执行下一条命令其中{...原创 2019-04-28 23:03:48 · 15631 阅读 · 3 评论