Keith

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

tensorflow学习笔记(四十三):使用 tfdbg 来 debug

由于 tensorflow 在训练的时候是在后台运行的,所以使用 python 的 debug 工具来 debug tensorflow 的执行过程是不可行的,为此,官方提供了一款debug 工具,名为 tfdbg 有很多人觉得,为了 debug tensorflow 的计算过程又要学习一个新的...

2017-08-22 14:14:13

阅读数 7039

评论数 3

一段代码搞懂 gpu memory

一段代码搞懂 gpu memoryGPU 的 memory 分为 三种,io速度从快到慢排序为: local memory shared memory global memory 其中 shared memory 的io 速度是远快于 global memory 的。这三种 memory 的访问性...

2017-08-16 09:52:33

阅读数 2852

评论数 1

pytorch: 如何优雅的将 int list 转成 one-hot形式

虽然 pytorch 已经升级到 0.2.0 了,但是,貌似依旧没有简单的 api 来帮助我们快速将 int list 转成 one-hot。那么,如何优雅的实现 one-hot 代码呢?def one_hot(ids, out_tensor): if not isinstance(ids...

2017-08-09 22:23:50

阅读数 9351

评论数 1

pytorch: 常见bug与debug

本博文用来记录自己的 pytorch 踩坑过程,虽然 pytorch 命令式编程,声称容易 debug,可是 代码给出的错误提示可是令人相当头疼,所以在本文中记录一下在编写 pytorch 代码过程中碰到的 坑,和如何 填坑。 TypeError: ××× received an invalid ...

2017-08-09 19:33:26

阅读数 6428

评论数 0

pytorch: Variable detach 与 detach_

pytorch 的 Variable 对象中有两个方法,detach和 detach_ 本文主要介绍这两个方法的效果和 能用这两个方法干什么。 detach 官方文档中,对这个方法是这么介绍的。 返回一个新的 从当前图中分离的 Variable。 返回的 Variable 永远不会需要梯度...

2017-08-05 12:57:55

阅读数 19983

评论数 10

tensorflow:使用tfrecords时的注意事项

使用 tfrecords 时的注意事项 确保 string_input_producer 中的文件名字是正确的。 string_input_producer(file_names, num_epochs=100000, shuffle=True) 当指定 num_epochs 时,在初始化模型参...

2017-08-05 12:09:01

阅读数 3312

评论数 1

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