随笔分类 -  深度学习~pytorch

pytorch调用的一些代码解读
摘要:1、pytorch+python optimizer=torch.optim.SGD(net.parameters(),lr=LR) #optimizer=torch.optim.SGD(net.parameters(),lr=LR,momentum=0.8) #optimizer=torch.op 阅读全文
posted @ 2020-04-21 11:23 皮卡皮卡妞 阅读(487) 评论(0) 推荐(0)
摘要:一、 卷积层 class torch.nn.Conv1d(in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=True) 一维卷积层,输入的尺度是(N, C_in,L),输出尺 阅读全文
posted @ 2020-02-21 10:05 皮卡皮卡妞 阅读(791) 评论(0) 推荐(0)
摘要:def tensor2im(image_tensor, imtype=np.uint8, normalize=True): if isinstance(image_tensor, list): image_numpy = [] for i in range(len(image_tensor)): i 阅读全文
posted @ 2020-02-18 14:50 皮卡皮卡妞 阅读(829) 评论(0) 推荐(0)