pytorch 记录

最新推荐文章于 2024-09-26 04:26:01 发布

原创最新推荐文章于 2024-09-26 04:26:01 发布 · 199 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch

pytorch 专栏收录该内容

3 篇文章

订阅专栏

部署运行你感兴趣的模型镜像

这里写自定义目录标题

pytorch 加载部分权重
pytorch 固定权重

pytorch 加载部分权重

pretrained_dict = torch.load('{0}/{1}'.format(opt.outf, opt.resume_posenet))
model_dict = model.state_dict()
#筛选条件
pretrained_dict =  {k: v for k, v in pretrained_dict.items() if k in model_dict}
model_dict.update(pretrained_dict)
model.load_state_dict(model_dict)

pytorch 固定权重

for p in self.parameters():
	p.requires_grad=False

您可能感兴趣的与本文相关的镜像

PyTorch 2.5

PyTorch

Cuda

PyTorch 是一个开源的 Python 机器学习库，基于 Torch 库，底层由 C++ 实现，应用于人工智能领域，如计算机视觉和自然语言处理

确定要放弃本次机会？

福利倒计时

: :

立减 ¥

普通VIP年卡可用

立即使用

FengGHT

关注关注

0
点赞
踩
0

收藏

觉得还不错? 一键收藏
0
评论
分享

复制链接

分享到 QQ

分享到新浪微博

扫一扫
举报

举报

专栏目录

python中format的用法

weixin_38410551的博客

09-25

385

参考链接： https://www.cnblogs.com/alan-babyblog/p/5227098.html

Python中format函数的用法

thy0000的博客

10-16

1057

str.format() 是 Python 字符串的格式化方法，用于将值插入字符串中的占位符。这个方法允许我们以更灵活的方式创建字符串，将变量、表达式或其他字符串值插入到已有字符串的特定位置。

参与评论您还未登录，请先登录后发表或查看评论

Python字符串格式化--format()方法

ZHOU的博客

10-31

2011

1.简单运用字符串类型格式化采用format()方法，基本使用格式是： <模板字符串>.format(<逗号分隔的参数>) 调用format()方法后会返回一个新的字符串，参数从0 开始编号。 print("{}：计算机{}的CPU 占用率为{}%。".format("2016-12-31","PYTHON",10)) output: '2016-12...

pytorch基础

zhanly19的博客

07-17

346

一、Tensor (张量) import torch from torch.autograd import Variable import numpy as np #tensor 张量 a = torch.Tensor([[2,3],[4,8],[7,9]]) print('a is:{}'.format(a)) print('a size is:{}'.format(a.size())) a...

pytorch笔记（一） ——pytorch数据操作

Leiroy的博客

05-07

439

文章目录1. 数据操作1.1创建Tensor1.2 运算1.3 广播机制1.4 索引1.5 运算的内存开销1.6 Tensor和NumPy相互变换 1. 数据操作 1.1创建Tensor 在深度学习中，我们通常会频繁地对数据进行操作。在PyTorch中， torch.Tensor 是一种包含单一数据类型元素的多维矩阵，也是存储和变换数据的主要工具。为了简洁，本书常Tensor 实例直接称作Tensor 。如果你之前用过NumPy ，你会发现Tensor 和NumPy 的多维数组非常类似。然而， T

Pytorch 记录每一次迭代的数据加载、模型计算等时长

weixin_45507687的博客

10-11

1180

使用 time 库如何记录pytorch处理每一批次时间消耗。可结合 tqdm 工具使用效果更佳。

PyTorch记录三：使用torch.nn构建自己的神经网络

liuxin19950129的博客

09-26

1266

如何自己定义一个神经网络

重装pytorch记录

m0_61175448的博客

11-15

964

之前在安装了python的情况下安装了anaconda，又装了pytorch，但是在pycharm上使用torchvision时总会出现warning，于是对自己的安装过程起疑，但是又想不起来具体细节，想起曾看到过说装anaconda之前最好不要装python，死马当活马医，先卸了python，再重新建个虚拟环境安一遍（似乎卸载anaconda挺麻烦，所以先试试看，还是不行再说）

ubuntu22.04安装nvidia驱动、cuda以及pytorch记录，亲测可行

chubbykkk的博客

02-27

3722

ubuntu22.04安装nvidia驱动、cuda以及pytorch记录，主要是为了纠正网上很多文章的错误，避免走弯路。我是在笔记本电脑测试的，安装了win11和ubuntu22.04双系统。笔记本配置是12代i7、16G内存、3060 6G 显卡。

docker 建立容器 - tensorflow & pytorch 记录

Elpecz

04-24

541

su docker run --gpus all -it -v /home:/home --name='base_' 3c2 /bin/bash apt-get update apt-get install vim vi /etc/apt/sources.list # 参考https://mirrors.tuna.tsinghua.edu.cn/help/ubuntu/ 换源 #安装miniCONDA https://docs.conda.io/en/latest/miniconda.html#lin

【PyTorch常用知识总结(一)】

syh的技术记录

11-28

2341

目录0. 前言1. 常用基本操作1.1 创建tensor1.2 tensor的基本数据类型:torch.dtype1.3 改变tensor的基本数据类型：tensor.type()1.3 改变tensor形状:tensor.view()1.4 获得tensor的某个元素的值 : .item()1.5 tensor和ndarray互相转换1.5.1 tensor ---> ndarray: tensor.numpy()1.5.2 ndarray ---> tensor: torch.from_n

Pytorch基础学习（第二章-Pytorch数据处理）

Billie使劲学的博客

03-29

3117

课程一览表：目录一、人民币二分类与DataSet与DataLoader 0.人民币二分类模型任务 1. DataLoader 2.Dataset 3.人民币二分类二、数据预处理transforms模块机制 1.transforms运行机制 2.数据标准化——transformes.normalize 三、transforms图像增强 1.什么是数据增强 2.transforms——裁剪 3.transforms——翻转和旋转 4.transforms——图像变换 5.

TSN算法的PyTorch代码解读（训练部分）

热门推荐

AI之路

01-14

4万+

这篇博客来读一读TSN算法的PyTorch代码，总体而言代码风格还是不错的，多读读优秀的代码对自身的提升还是有帮助的，另外因为代码内容较多，所以分训练和测试两篇介绍，这篇介绍训练代码，介绍顺序为代码运行顺序。TSN算法的介绍可以参考博客TSN(Temporal Segment Networks)算法笔记。论文：Temporal Segment Networks: Towards Good Pr

深度学习入门之Pytorch——Momentum

weixin_40793406的博客

11-30

6457

动量法动量法是梯度下降法的变式，在随机梯度下降的同时，增加动量。这是来自于物理中的概念，可以想象损失函数是一个山谷，一个球从山谷滑下来，在一个平坦的地势，球的滑动速度就会慢下来，可能陷入一些鞍点或者局部极小值点，如图这个时候给它增加动量就可以让它从高处滑落时的势能转换为平地的动能，相当于惯性增加了小球在平地滑动的速度，从而帮助其跳出鞍点或者局部极小点。动量怎么计算呢？动量的计算基于前面的梯...

Pytorch简单入门操作

m0_61899108的博客

02-10

1450

前言本文概述pytorch的一些简单操作。摘取于笔记 | PyTorch安装及入门教程 PyTorch主要由4个包组成： torch:可以将张量转换为torch.cuda.TensorFloat torch.autograd:自动梯度 torch.nn：具有共享层和损失函数的神经网络库 torch.optim：具有通用的优化算法包（SGD、Adam）目录前言 1 Numpy 与 Tensor 1.1 Tensor概述 1.2 创建Tensor

PyTorch官方教程之4：PyTorch之数据加载和处理

菜鸟笔记

03-14

1822

# 《PyTorch官方教程中文版》， PyTorch之数据加载和处理 from __future__ import print_function, division import os import torch import pandas as pd # 用于更容易地进行csv解析 from skimage import io, transform # 用于图像的IO和变换 import numpy as np import matplotlib.pyplot as plt from torch.ut

pytorch 张量

weixin_42528089的博客

12-05

1268

例程1: import torch t2=torch.tensor([[0,1,2],[3,4,5]]) print(t2) print('数据={}'.format(t2)) print(t2.reshape(3,2)) print(t2+1) print('大小={}'.format(t2.size())) print('维度={}'.format(t2.dim())) print('元素...

pytorch学习

lantuxin的博客

02-19

5407

1.pytorch环境配置(docker) docker环境--配置过程(配置成功--能使用nvidia-docker命令) 拉取pytorch镜像--参考网站(命令--docker pull nvcr.io/nvidia/pytorch:19.01-py3) 启动docker容器--参考网站(命令--sudo nvidia-docker run -it --rm -v local_dir:...

pytorch之卷积模块、池化、激活函数（可视化）

朴素.无恙的博客

10-29

1万+

pytorch之卷积模块卷积卷积在 pytorch 中有两种方式，一种是 torch.nn.Conv2d()，一种是torch.nn.functional.conv2d()，这两种形式本质都是使用一个卷积操作这两种形式的卷积对于输入的要求都是一样的，首先需要输入是一个 torch.autograd.Variable() 的类型，大小是 (batch, channel, H, W)，其中 bat...

pytorch