随笔分类 -  pytorch+pytorch linghtling

摘要:total_num = sum(p.numel() for p in model.parameters()) trainable_num = sum(p.numel() for p in model.parameters() if p.requires_grad) print('Total ',to 阅读全文
posted @ 2021-03-17 12:08 水木清扬 阅读(580) 评论(0) 推荐(1)
摘要:AI编辑:我是小将 本文作者:白裳 https://zhuanlan.zhihu.com/p/145842317 本文已由原作者授权 目前 pytorch 已经在 torchvision 模块集成了 FasterRCNN 和 MaskRCNN 代码。考虑到帮助各位小伙伴理解模型细节问题,本文分析一下 阅读全文
posted @ 2021-02-01 13:51 水木清扬 阅读(2096) 评论(0) 推荐(0)
摘要:Pytorch中的Distributed Data Parallel与混合精度训练(Apex) 磐创AI 昨天 以下文章来源于Smarter ,作者薰风初入弦 Smarter Smarter做CV界最优质的内容输出 磐创AI推荐搜索 Keras PyTorch Github 磐创AI分享 来源 | 阅读全文
posted @ 2021-01-08 10:43 水木清扬 阅读(1455) 评论(0) 推荐(0)
摘要:易于使用的神经风格迁移框架 pystiche。 将内容图片与艺术风格图片进行融合,生成一张具有特定风格的新图,这种想法并不新鲜。早在 2015 年,Gatys、 Ecker 以及 Bethge 开创性地提出了神经风格迁移(Neural Style Transfer ,NST)。不同于深度学习,目前 阅读全文
posted @ 2020-12-23 09:45 水木清扬 阅读(752) 评论(0) 推荐(0)
摘要:Pytorch中的tensor又包括CPU上的数据类型和GPU上的数据类型,一般GPU上的Tensor是CPU上的Tensor加cuda()函数得到。 一般系统默认是torch.FloatTensor类型。例如data = torch.Tensor(2,3)是一个2*3的张量,类型为FloatTen 阅读全文
posted @ 2020-12-22 08:56 水木清扬 阅读(3013) 评论(0) 推荐(0)
摘要:OpenMMLab 机器学习算法工程师 今天 AI编辑:我是小将 本文作者: OpenMMLab https://zhuanlan.zhihu.com/p/321449610 本文已由原作者授权 前言 本篇笔记以介绍 pytorch 中的 autograd 模块功能为主,主要涉及 torch/aut 阅读全文
posted @ 2020-12-21 13:41 水木清扬 阅读(1434) 评论(0) 推荐(0)
摘要:PyTorch 源码解读之 BN & SyncBN 机器学习算法工程师 今天 点蓝色字关注“机器学习算法工程师” 设为星标,干货直达! AI编辑:我是小将 本文作者:OpenMMLab @205120 https://zhuanlan.zhihu.com/p/337732517 本文已由原作者授权 阅读全文
posted @ 2020-12-21 13:38 水木清扬 阅读(1975) 评论(0) 推荐(0)
只有注册用户登录后才能阅读该文。
posted @ 2019-01-23 20:53 水木清扬 阅读(4) 评论(0) 推荐(0)
只有注册用户登录后才能阅读该文。
posted @ 2019-01-23 16:16 水木清扬 阅读(8) 评论(0) 推荐(0)
摘要:读取图片数据集::# -*- coding: utf-8 -*-import torch as tfrom torch.utils import dataimport osfrom PIL import Imageimport numpy as npclass DogCat(data.Dataset 阅读全文
posted @ 2019-01-23 15:00 水木清扬 阅读(533) 评论(0) 推荐(0)
摘要:2018年07月07日 17:30:40 __矮油不错哟 阅读数:221 2018年07月07日 17:30:40 __矮油不错哟 阅读数:221 1. 数据处理 数据加载 ImageFolder DataLoader加载数据 sampler:采样模块 1. 数据处理 数据加载 ImageFolde 阅读全文
posted @ 2019-01-23 14:52 水木清扬 阅读(2693) 评论(0) 推荐(0)
摘要:# -*- coding: utf-8 -*-"""View more, visit my tutorial page: https://morvanzhou.github.io/tutorials/My Youtube Channel: https://www.youtube.com/user/M 阅读全文
posted @ 2019-01-18 16:57 水木清扬 阅读(242) 评论(0) 推荐(0)
摘要:# -*- coding: utf-8 -*-import torchimport matplotlib.pyplot as plt# torch.manual_seed(1) # reproducible# fake data从-1到1分成100分x = torch.unsqueeze(torch 阅读全文
posted @ 2019-01-18 16:46 水木清扬 阅读(201) 评论(0) 推荐(0)
摘要:# -*- coding: utf-8 -*-"""View more, visit my tutorial page: https://morvanzhou.github.io/tutorials/My Youtube Channel: https://www.youtube.com/user/M 阅读全文
posted @ 2019-01-18 15:39 水木清扬 阅读(296) 评论(0) 推荐(0)
只有注册用户登录后才能阅读该文。
posted @ 2019-01-17 13:52 水木清扬 阅读(6) 评论(0) 推荐(0)
只有注册用户登录后才能阅读该文。
posted @ 2019-01-16 15:37 水木清扬 阅读(6) 评论(0) 推荐(0)
摘要:目录 Pytorch Leture 05: Linear Rregression in the Pytorch Way Logistic Regression 逻辑回归 - 二分类 Lecture07: How to make netural network wide and deep ? Lect 阅读全文
posted @ 2019-01-16 14:26 水木清扬 阅读(561) 评论(0) 推荐(0)
摘要:Pytorch学习系列(一)至(四)均摘自《深度学习框架PyTorch入门与实践》陈云目录:1.程序的主要功能2.文件组织架构3. 关于`__init__.py`4.数据处理5.模型定义6.工具函数7.配置文件8.main.py9.使用1.程序的主要功能: 模型定义 数据加载 训练和测试2.文件组织 阅读全文
posted @ 2019-01-14 12:03 水木清扬 阅读(3274) 评论(0) 推荐(0)
只有注册用户登录后才能阅读该文。
posted @ 2019-01-14 12:01 水木清扬 阅读(4) 评论(0) 推荐(0)
只有注册用户登录后才能阅读该文。
posted @ 2019-01-14 11:59 水木清扬 阅读(2) 评论(0) 推荐(0)