随笔分类 - 深度学习~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
阅读全文
摘要:一、 卷积层 class torch.nn.Conv1d(in_channels, out_channels, kernel_size, stride=1, padding=0, dilation=1, groups=1, bias=True) 一维卷积层,输入的尺度是(N, C_in,L),输出尺
阅读全文
摘要:def tensor2im(image_tensor, imtype=np.uint8, normalize=True): if isinstance(image_tensor, list): image_numpy = [] for i in range(len(image_tensor)): i
阅读全文

浙公网安备 33010602011771号