![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
小清新嘻嘻无敌
这个作者很懒,什么都没留下…
展开
-
pytorch deeplearning测试阶段GPU测试问题
按照https://www.jianshu.com/p/889dbc684622实现了CIFAR10的分类,但是转到GPU上之后,出现error,是数据类型转换问题。原在cpu上测试的代码:correct = 0total = 0for data in testloader: images, labels = data outputs = net(Variable...原创 2018-08-03 16:50:08 · 1266 阅读 · 0 评论 -
pytorch学习-莫烦
PyTorch 官网https://www.bilibili.com/video/av15997678/?p=14https://morvanzhou.github.io/tutorials/machine-learning/torch/3-04-save-reload/要点 训练好了一个模型, 我们当然想要保存它, 留到下次要用的时候直接提取直接用, 这就是这节的内容啦. 我们用回...转载 2018-08-17 11:54:37 · 673 阅读 · 0 评论 -
torch.normal()
torch.normal()torch.normal(means, std, out=None)返回一个张量,包含从给定参数means,std的离散正态分布中抽取随机数。 均值means是一个张量,包含每个输出元素相关的正态分布的均值。 std是一个张量,包含每个输出元素相关的正态分布的标准差。 均值和标准差的形状不须匹配,但每个张量的元素个数须相同。参数:means (Ten...转载 2018-08-17 12:43:35 · 73890 阅读 · 16 评论 -
Pytorch指定使用的GPU
转自 http://www.cnblogs.com/darkknightzh/p/6836568.html PyTorch默认使用从0开始的GPU,如果GPU0正在运行程序,需要指定其他GPU。有如下两种方法来指定需要使用的GPU。1. 类似tensorflow指定GPU的方式,使用CUDA_VISIBLE_DEVICES。1.1 直接终端中设定:CUDA_VISIBLE...转载 2018-08-17 14:57:57 · 1783 阅读 · 0 评论 -
pytorch 广播语义(Broadcasting semantics)
许多PyTorch操作都支持。NumPy Broadcasting Semantics简而言之,如果PyTorch操作支持广播,那么它的Tensor参数可以自动扩展为相同的大小(不需要复制数据)。一般语义如果遵守以下规则,则两个张量是“可播放的”:每个张量至少有一个维度。 迭代尺寸大小时,从尾随尺寸开始,尺寸大小必须相等,其中一个为1,或者其中一个不存在。例如:>&...转载 2018-09-07 10:00:55 · 4174 阅读 · 0 评论 -
pytorch copy_
copy_(src, non_blocking=False) → TensorCopies the elements from src into self tensor and returns self.The src tensor must be broadcastable with the self tensor. It may be of a different data type ...原创 2018-09-07 10:21:52 · 5141 阅读 · 0 评论