首页
学习
活动
专区
圈层
工具
发布

#torch

MysticMirror 自研大模型全维度自动化测试套件(42 项用例)

王瑞MVP

本文档为 MysticMirror 自研大模型的全覆盖自动化回归测试套件,整合原有 22 项基础回归测试与 20 项新增边界能力测试,共计 42 个测试用例。全...

3910

大模型网络结构:模型接口测试用例参考

王瑞MVP

大模型的网络文件是模型训练的核心参考。例如 MiniMind 项目中 /model/model_minimind.py 这份网络结构文件,它用来定义整套大模型的...

13110

Nunchaku Windows 无需预编译直接下载whl文件与安装教程(Torch 2.7 / 2.8)

代码简单说

前段时间在本地折腾 Nunchaku 的 Windows 环境,踩了不少坑,尤其是 CUDA + Torch + Python 版本 对不齐的问题。 后来整理...

58610

CacheDiT、TaylorSeer 与 SCM:DiT 扩散模型推理加速到底在加速什么?

Michael阿明

在图像、视频生成模型里, Diffusion Transformer,简称 DiT

38710

一招教你 解决torch.hub.load模型加载问题

OpenCV学堂

PyTorch Hub 提供了方便的 API,通过 torch.hub.list() 查询所有可用的模型;通过 torch.hub.help() 显示文档字符串...

45710

从零理解Softmax在分类问题作用

索旭东

假设你训练了一个神经网络,让它识别图像中是 猫、狗还是鸟 (3分类问题)。网络的最后一层通常有3个神经元,每个输出一个数值,我们称这些数值为 logits (原...

87300

从0开始训练自己的LLM(2)

golangLeetcode

Transformer架构是深度学习领域的革命性设计,由谷歌大脑团队在2017年提出,通过自注意力机制彻底改变了序列建模方式。其核心优势在于并行...

25310

torch.nn.Embedding 中 max_norm 的作用

AlphaHinex

nn.Embedding(num_embeddings, embedding_dim) 可以看成是一个查表词典:

23610

torch.compile 加速原理:kernel 融合与缓冲区复用

deephub

torch.compile 通过提前构建计算图来解决这个问题。它的核心策略是操作融合和缓冲区复用:第一次调用需要编译而之后的推理会快很多。在 PyTorch 官...

43410

30:【uv + torch】2026最快安装GPU PyTorch & 避开conda坑

安全风信子

摘要: 本文详细分析2026年使用uv和torch快速安装GPU版本PyTorch的方法,以及如何避开conda的常见问题。文章提供了完整的安装指南、性能对比、...

3K10

卷积神经网络

贺公子之数据科学与艺术

卷积神经网络(CNN)主要由输入层、卷积层、激活函数、池化层和全连接层组成。典型结构为:

71010

从零实现3D Gaussian Splatting:完整渲染流程的PyTorch代码详解

deephub

3D Gaussian Splatting(3DGS)现在几乎成了3D视觉领域的标配技术。NVIDIA把它整合进COSMOS,Meta的新款AR眼镜可以直接在设...

1.5K10

Transformer自回归关键技术:掩码注意力原理与PyTorch完整实现

deephub

掩码注意力(Causal Attention)是生成式模型的核心技术,它传统自注意力机制有根本的不同,掩码注意力限制模型只能关注当前位置之前的tokens,确保...

88510

深入解析torch.compile:提升PyTorch模型性能、高效解决常见问题

deephub

近年来,PyTorch已在学术界和工业界稳固了其作为主流深度学习框架的地位。随着PyTorch 2.0的发布,其核心功能之一torch.compile为用户提供...

2.3K10

AI如何理解语言?自注意力机制的技术原理与代码实践

AI浩

当你与ChatGPT、文心一言或通义千问对话时,是否曾好奇过:AI是如何理解你的话语并给出连贯回复的?为什么它能抓住句子中词语间的微妙关系,甚至能领会言外之意?

60110

导入apex报错AttributeError: module ‘torch‘ has no attribute ‘library‘

git clone firc-dataset

always have Traceback (most recent call last): File "", line 1, in File "/roo...

51300

[环境配置]stable diffusion启动报错cannot import name ‘get_device‘ from ‘basicsr.utils.misc‘

git clone firc-dataset

python webui.py启动后报错cannot import name 'get_device' from 'basicsr.utils.misc'

25010

[深度学习][CenterFusion]关于centerfusion训练注意的要点

git clone firc-dataset

pytorch版本问题:我使用的torch==1.11.0+cu113 torchvision==0.12.0+cu113版本,训练时候会报错torchvisi...

24400

[python]torch.cat和numpy.concatenate对应拼接

git clone firc-dataset

因此torch.cat函数和Numpy中concatenate对应,numpy里面是没有cat函数的

33010

[pytorch]pytorch官方安装法

git clone firc-dataset

conda install pytorch=0.4.1 cuda90 -c pytorch conda install pytorch=0.4.1 cuda92...

1.7K10
领券