pytorch学习
aiqiu_gogogo
中国科学院自动化研究所研究生,臭淋雨,坑货一枚,萌萌的,么么哒~~~
展开
-
计算一下pytorch中Resnet34模型前传一次所需要的时间
import pdbimport timeimport torchimport torchvision.models as modelsfrom torch.autograd import Variableclass Timer(object): """A simple timer.""" def __init__(self): self.total_time原创 2017-12-03 20:53:15 · 2638 阅读 · 0 评论 -
纪念一下终于搞出来的PyTorch的Multi-GPU使用代码
PyTorch多GPU使用例程import osimport pdbimport timeimport torchimport torch.nn as nnfrom torch.autograd import Variableclass Timer(object): """A simple timer.""" def __init__(self): self原创 2017-11-26 10:43:48 · 14939 阅读 · 2 评论 -
我在读pyTorch文档(一)
Cuda在Cuda上创建变量的两个方法:直接在GPU上创建:x = torch.cuda.FloatTensor(1);在CPU上创建然后转移到GPU上:x = torch.FloatTensor(1).cuda();多GPU使用:x = torch.FloatTensor(1).cuda(async=True), 通过async=True可以将数据从CPU到GPU的传输与计算重叠,不过当数据原创 2017-11-26 11:53:54 · 1165 阅读 · 0 评论 -
我在读pyTorch文档(二)
常用Torch自带Tensor处理函数bool = torch.is_tensor(x):判断x是否为pytorch张量;bool = torch.is_storage(x):判断x是否为pytorch对象;N = torch.numel(x):输出pytorch张量x的元素数;x = torch.eye(n):输出n*n的二维单位矩阵;y = torch.from_numpy(x):将n原创 2017-11-27 09:50:55 · 9130 阅读 · 0 评论 -
纪念一下pyTorch自由读取txt路径存储格式的图片数据
pyTorch自由读取txt路径存储格式的图片数据import pdbimport numpy as npimport torch, cv2, os, randomfrom torch.utils import datafrom torchvision import transformsclass TxtImage(data.Dataset): def __init__(self, la原创 2017-11-27 10:52:16 · 3270 阅读 · 1 评论 -
我在读pyTorch文档(三)
数据类型CPU数据类型:torch.FloatTensor(torch.Tensor),torch.DoubleTensor,torch.ByteTensor,torch.CharTensor,torch.ShortTensor,torch.IntTensor,torch.LongTensor;GPU数据类型:torch.cuda.FloatTensor,torch.cuda.DoubleTen原创 2017-11-27 15:22:37 · 896 阅读 · 0 评论 -
我在读pyTorch文档(四)
torch.nn.Module打印所有子模块:for sub_module in model.children(): print(sub_module)按照名字打印子模块:for name, module in model.named_children(): if name in ['conv4', 'conv5']: print(module)打印所有模块:for原创 2017-11-27 19:31:31 · 2679 阅读 · 0 评论 -
好的pytorch网络资源
持续更新。。。。。。。。。。。。。。。。实现Siamese:https://hackernoon.com/facial-similarity-with-siamese-networks-in-pytorch-9642aa9db2f7原创 2017-11-27 21:36:24 · 1099 阅读 · 0 评论