![](https://img-blog.csdnimg.cn/20201014180756754.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
Pytorch
文章平均质量分 91
莱维贝贝、
这个作者很懒,什么都没留下…
展开
-
PyTorch自定义损失函数、动态调整学习率、模型微调、半精度训练
PyTorch进阶训练技巧import torchimport numpy as npimport torch.nn as nnimport torch.nn.functional as F1. 自定义损失函数# 以函数的方式定义损失函数,通过输出值和目标值进行计算,返回损失值def my_loss(output,target): loss = torch.mean((output - target)**2) return lossy_pred = torch.tenso原创 2022-03-21 15:01:41 · 3507 阅读 · 0 评论 -
Pytorch定义模型、修改模型、保存与读取模型保存
1. PyTorch的模型定义1.1 PyTorch模型定义的方式PyTorch中有三种模型定义方式,三种方式都是基于nn.Module建立的,我们可以通过Sequential,ModuleList和ModuleDict三种方式定义PyTorch模型。Module类是torch.nn模块里提供的一个模型nn.Module,是所有神经网络的基础模型:1.1.1 Sequential优点: 简单、易读,同时使用Sequential定义的模型不需要再写forward,因为顺序已经定义好了。缺点:原创 2022-03-16 17:44:51 · 7310 阅读 · 2 评论 -
Ubuntu18.04环境下安装Anaconda3+pytorch
1. Ubuntu 卸载与安装 Anaconda31.1 Ubuntu 卸载 Anaconda3第一步:删除Anaconda3文件夹~ : 表示你的/home/用户明目录例如:假设你的用户名是bianbian,那么 ~/ 就是 /home/bianbian/我的anaconda3在用户目录下,~/anaconda3 = /home/bianbian/anaconda3rm -rf ~/anaconda31.2 删除Anaconda·配置的环境变量 sudo vim ~/.bashrc原创 2022-03-13 17:00:31 · 4935 阅读 · 0 评论