pytorch
文章平均质量分 92
3阿谭
这个作者很懒,什么都没留下…
展开
-
可视化
首先把tensorboardX导入并进行初始化:from tensorboardX import SummaryWriter# 定义Summary_Writerwriter = SummaryWriter('./Result') # 数据存放在这个文件夹基础函数add_scalar()add_scalar(tag, scalar_value, global_step=None, ...原创 2019-05-20 15:43:50 · 577 阅读 · 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 · 359 阅读 · 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 · 230 阅读 · 0 评论 -
损失函数与优化器
十七个损失函数L1lossclass torch.nn.L1Loss(size_average=None, reduce=None)计算 output 和 target 之差的绝对值,可选返回同维度的 tensor 或者是一个标量。MSELossclass torch.nn.MSELoss(size_average=None, reduce=None, reduction='elemen...原创 2019-05-18 19:11:59 · 2407 阅读 · 0 评论