
pytorch
文章平均质量分 92
3阿谭
这个作者很懒,什么都没留下…
展开
专栏收录文章
- 默认排序
- 最新发布
- 最早发布
- 最多阅读
- 最少阅读
-
可视化
首先把tensorboardX导入并进行初始化: from tensorboardX import SummaryWriter # 定义Summary_Writer writer = SummaryWriter('./Result') # 数据存放在这个文件夹 基础函数 add_scalar() add_scalar(tag, scalar_value, global_step=None, ...原创 2019-05-20 15:43:50 · 642 阅读 · 0 评论 -
数据
划分 训练集(train set)、验证集(valid/dev set)和测试集(test set) = 随堂测试、模拟考、高考 加载流程 1.数据预处理 2.构建MyDataset实例 3.构建Dataloader 数据预处理 PyTorch 做数据增强的方法是在原始图片上进行的,并覆盖原始图片。 # 设置均值,标准差,以及数据标准化 normMean = [0.4948052, 0.48568...原创 2019-05-17 15:24:22 · 418 阅读 · 0 评论 -
模型
模型的定义 首先,必须继承 nn.Module 类。 其次,在__init__(self)中设置好需要的“组件"(如 conv、 pooling、 Linear、 BatchNorm等)。 最后,在 forward(self, x)中用定义好的“组件”进行组装。 class Net(nn.Module): def __init__(self): # 初始化,定义组件 ...原创 2019-05-17 18:51:08 · 267 阅读 · 0 评论 -
损失函数与优化器
十七个损失函数 L1loss class torch.nn.L1Loss(size_average=None, reduce=None) 计算 output 和 target 之差的绝对值,可选返回同维度的 tensor 或者是一个标量。 MSELoss class torch.nn.MSELoss(size_average=None, reduce=None, reduction='elemen...原创 2019-05-18 19:11:59 · 2497 阅读 · 0 评论