pytorch
沐雨金鳞
大行不顾细谨
展开
-
神经网络中的一些小东西
1、argparse import argparse parser = argparse.ArgumentParser(description='learn the argparse') parser.add_argument('--lr', default=0.1, type=float, help='learning rate') parser.add_argument('--epoc...原创 2019-12-16 16:22:40 · 309 阅读 · 0 评论 -
卷积操作和全连接操作顺序
卷积层:conv+relu+batch_normalize+pooling 全连接层:Linear+relu+dropout原创 2019-12-12 16:07:14 · 608 阅读 · 0 评论 -
pytorch实战3
1、tensor<--->numpy import torch import numpy as np x = np.array([[1,2],[3,4]]) # 将numpy换成torch tensor y = torch.from_numpy(x) # 将torch tensor换成numpy z = y.numpy() 2、数据下载 import torch imp...原创 2019-10-13 15:02:59 · 238 阅读 · 0 评论 -
pytorch实战2
1、利用torch.nn包中的误差函数 import torch from torch.autograd import Variable # 批量输入的数据量 batch_n = 100 # 通过隐藏层后输出的特征数 hidden_layer = 100 # 输入数据的特征个数 input_data = 1000 # 最后输出的分类结果数 output_data = 10 x = Vari...原创 2019-10-13 14:59:19 · 140 阅读 · 0 评论 -
pytorch实战1
1、Tensor的数据类型 # torch.FloatTensor浮点型 a = torch.FloatTensor(2,3) print(a) b = torch.FloatTensor([2,3,4,5]) print(b) # torch.IntTensor整型 a = torch.IntTensor(2,3) print(a) b = torch.IntTensor([2,3,4,5...原创 2019-10-13 14:54:21 · 512 阅读 · 1 评论