Keith

记录与分享学习点滴。 QQ:862820505 (有偿咨询)

pytorch学习笔记(十六):pytorch 写代码时应该注意

当网络中有 dropout,bn 的时候。训练的要记得 net.train(), 测试 要记得 net.eval() 在测试的时候 创建输入 Variable 的时候 要记得 volatile=True torch.sum(Tensor), torch.mean(Tensor) 返回的是 pyth...

2017-11-24 13:14:08

阅读数 1956

评论数 0

C 中内存的一些操作

分配内存与释放内存 在C 中,下列函数的声明在 stdlib.h 中,在 C++, 声明在 cstdlib 中malloc realloc calloc free , 释放分配的内存(分配的内存块的头部记录了当前分配内存块的大小) 一、 malloc malloc(memory alloc...

2017-11-19 23:37:57

阅读数 295

评论数 0

pytorch学习笔记(十五):pytorch 源码编译碰到的坑总结

2017.11.17最近打算学习一下 pytorch 源码,所以按照官网的教程从头编译了一下 pytorch 。在编译的过程中,碰到了两个坑,在这里记录一下。 第一个错误 CMakeFiles/Makefile2:264: recipe for target 'src/ATen/test/CMak...

2017-11-17 09:10:19

阅读数 6452

评论数 0

pytorch学习笔记(十四): DataLoader源码阅读

pytorch 数据加载部分的 接口可以说是现存 深度学习框架中设计的最好的, 给了我们足够的灵活性。本博文就对 pytorch 的多线程加载 模块(DataLoader) 进行源码上的注释。 输入流水线 pytorch 的输入流水线的操作顺序是这样的: 创建一个 Dataset 对...

2017-11-15 22:18:59

阅读数 16089

评论数 3

pytorch学习笔记(十三):backward过程的底层实现解析

博主水平有限,如有错误,请不吝指出。 pytorch源码注释,欢迎 pr,提 issue 和 star 当我们使用 pytorch 的 python 的接口编写代码的时候,感觉是十分清爽的,不需要考虑底层的实现。但是好奇心驱使我们 想一探究竟,看看底层 C/C++ 那部分到底做了什么。 本篇...

2017-11-12 12:05:53

阅读数 4496

评论数 1

提示
确定要删除当前文章?
取消 删除
关闭
关闭