![](https://img-blog.csdnimg.cn/20201014180756919.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
文章平均质量分 91
赵YN的csdn
这个作者很懒,什么都没留下…
展开
-
pytorch中torch.Tensor.scatter用法
今天读代码,看到有人用了torch.Tensor.scatter这个函数。这个函数之前我也看到过,但是没有搞明白是干啥用的,今天我搞明白了。首先看一下官方文档的定义:(可跳过)==============================pytorch docs分割线====================================scatter_(dim, index, src)...原创 2019-07-16 11:30:29 · 13522 阅读 · 5 评论 -
使用torchsnooper自动输出变量的类型(float/int , cpu/cuda)
又来更新啦!要想不被生活按在地上摩擦,还是得功夫下在平时!今天更新一个发现的好包,此处应有掌声(啪啪啪)--> torchsnooper也是属于我pytorch大军中的一员,哈哈。这个主要是解决什么问题内?不知道你们会不会遇到这种情况:RuntimeError: Expected object of backend CPU but got backend CUDA for...原创 2019-07-24 19:59:04 · 920 阅读 · 0 评论 -
python cuda cpu中随机种子(一)
最近一直在做A3C框架中的强化学习算法,发现随机种子的设定非常重要。首先,为了代码的可复现性,要设定随机种子,让每次代码跑出来的结果一样。其次,由于A3C框架的特殊性,每个worker(一个子线程)之间要保证其多样性。但是,我在设定完了之后,发现跑出来的结果并不能保证完全一模一样,于是对于seed做了一些实验,得出了一点心得。首先,说一下几个随机种子:python&numpy:...原创 2019-08-11 17:23:02 · 1247 阅读 · 2 评论