![](https://img-blog.csdnimg.cn/20201014180756926.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
Pytorch
文章平均质量分 96
Pytorch基础应用
神洛华
这个作者很懒,什么都没留下…
展开
-
PyTorch官方文档——CV篇上:Mask R-CNN实例分割,ConvNet图片分类,对抗训练FGSM
翻译了PyTorch官方文档Image and Video部分教程翻译 2022-10-11 04:57:03 · 860 阅读 · 0 评论 -
yolov5官网教程&kaggle海星目标检测赛&paddle钢铁缺陷检测赛(VOC格式)
一是用yolov5s V6.1版本训练PASCAL VOC2012数据集,二是稍微翻译了下yolov5官网教程Train Custom Data。亲测通过,中间各种报错、配置文件、改路径都写得很清楚。原创 2022-08-11 00:59:17 · 3211 阅读 · 3 评论 -
PyTorch学习笔记1——基本概念、模块简介、张量操作、自动微分
张量的创建方式:shape = (2,3,)rand_tensor = torch.rand(shape)ones_tensor = torch.ones(shape)zeros_tensor = torch.zeros(shape)x_np = torch.from_numpy(np_array)print(f"Shape of tensor: {tensor.shape}")print(f"Datatype of tensor: {tensor.dtype}")print(f"Dev原创 2021-09-27 01:01:24 · 843 阅读 · 0 评论 -
编写transformers的自定义pytorch训练循环(Dataset和DataLoader解析和实例代码)
1.torch.utils.datatorch.utils.data主要包括以下三个类:class torch.utils.data.Dataset其他的数据集类必须是torch.utils.data.Dataset的子类,比如说torchvision.ImageFolder.class torch.utils.data.sampler.Sampler(data_source)参数: data_source (Dataset) – dataset to sample from作用: 创建一个采原创 2021-09-25 12:11:25 · 4761 阅读 · 0 评论 -
PyTorch学习笔记2:nn.Module、优化器、模型的保存和加载、TensorBoard
一、nn.Module1.1 nn.Module的调用pytorch通过继承nn.Module类,定义子模块的实例化和前向传播,实现深度学习模型的搭建。其构建代码如下:import torchimport torch.nn as nnclass Model(nn.Module): def __init__(self, *kargs): # 定义类的初始化函数,...是用户的传入参数 super(Model, self).__init__()#调用父类nn.Module的初原创 2021-11-18 00:39:41 · 12961 阅读 · 1 评论