首页
学习
活动
专区
工具
TVP
发布
精选内容/技术社群/优惠产品,尽在小程序
立即前往

#torch

pytorch之autograd

Tom2Code

我们都知道pytorch是经典的动态图,所以这个参数retain_graph是一个布尔类型的值,它的true or false直接说明了在计算过程中是否保留图

300

【踩坑】PyTorch中指定GPU不生效和GPU编号不一致问题

小锋学长生活大爆炸

解释:就是使用os.environ["CUDA_VISIBLE_DEVICES"] = "1"后,后面使用起来仍然是cuda0.

2300

JetBot手势识别实验

zstar

本实验目的在JetBot智能小车实现手势识别功能,使用板卡为Jetson Nano。通过小车摄像头,识别五个不同的手势,实现小车的运动及灯光控制。

4400

如何解决ModuleNotFoundError: No module named ‘torch’问题

猫头虎

本文旨在向广大开发者和热爱机器学习的同学们介绍如何解决在Python环境中常见的错误ModuleNotFoundError: No module named ‘...

4100

250行代码从头搭建Llama 3,GitHub一天4.6k星!Karpathy大赞

新智元

再使用PyTorch内置的神经网络模块(torch.nn)将token转换为embedding,17x1的token维度变为17x4096。

8610

Karpathy称赞,从零实现LLaMa3项目爆火,半天1.5k star

机器之心

一个月前,Meta 发布了开源大模型 llama3 系列,在多个关键基准测试中优于业界 SOTA 模型,并在代码生成任务上全面领先。

7510

树莓派安装pytorch环境记录

鳄鱼儿

下载有点慢,可以先本地下载好github.com/KumaTea/pyt…,再离线安装

9910

flash-linear-attention的fused_recurrent_rwkv6 Triton实现精读

BBuf

继续补 在GPU上加速RWKV6模型的Linear Attention计算 没有写完的内容,对flash-linear-attention库(https://g...

6010

使用PyTorch实现L1, L2和Elastic Net正则化

数据STUDIO

L1 正则化的特点是它可以产生稀疏模型,即许多模型参数会被设置为零。这种特性使得L1正则化不仅可以防止过拟合,还可以进行特征选择。

13610

PyTorch的卷积和池化

@小森

卷积的计算是将卷积核放入左上角,在局部区域间做点积,然后将卷积核在Input上面依次从左向右,从上到下平移。左上角的点积操作:

7710

【实测】小白一看就学会的python的AI模型torch(中)

我去热饭

在python的AI库torch的官方教程中是明确可以支持多维度数据的。所以应该可以继续使用。那么代码中get_test_data就要变成如下的:

6610

【实测】小白一看就学会的python的AI模型torch(上)可以预测你的结婚年龄!!!

我去热饭

最近正好在攻克AI用例生成的技术,所以恰好弄到了这个知识点,踩了个坑,觉得挺有意义,就特此记录一下~

6510

使用PyTorch实现L1, L2和Elastic Net正则化

deephub

在机器学习中,L1正则化、L2正则化和Elastic Net正则化是用来避免过拟合的技术,它们通过在损失函数中添加一个惩罚项来实现。

7310

已解决Error | PyTorch下载慢:TimeoutError: [Errno 110] Connection timed out while downloading torch-猫头虎

猫头虎

你是否也遇到了在尝试从PyTorch官方网站下载时,面临下载速度缓慢甚至超时的问题?😿 在本文中,我——猫头虎博主,将带你深入了解这个问题的原因,并提供一套详尽...

31810

【踩坑】libtorch load 报错 No such file or directory

小锋学长生活大爆炸

你下的libtorch是release版本的,所以你在使用的时候,也应当使用release编译,不然就会报错。

10710

Pytorch-自动微分模块

@小森

自动微分模块是PyTorch中用于实现张量自动求导的模块。PyTorch通过torch.autograd模块提供了自动微分的功能,这对于深度学习和优化问题至关重...

9410

Pytorch-张量形状操作

@小森

在搭建网络模型时,掌握对张量形状的操作是非常重要的,因为这直接影响到数据如何在网络各层之间传递和处理。网络层与层之间很多都是以不同的 shape 的方式进行表现...

9410

医学图像处理案例(二十四)——基于cuda小波变换和cuda脉冲耦合神经网络的图像融合

医学处理分析专家

图像融合(Image Fusion)是指将多源信道所采集到的关于同一目标的图像数据经过图像处理和计算机技术等,最大限度的提取各自信道中的有利信息,最后综合成高质...

14621

医学图像处理案例(二十二)——基于cuda的小波变换的图像融合

医学处理分析专家

图像融合(Image Fusion)是指将多源信道所采集到的关于同一目标的图像数据经过图像处理和计算机技术等,最大限度的提取各自信道中的有利信息,最后综合成高质...

13110

Win环境下安装 torch==1.1.0

JOYCE_Leo16

复现很多模型的时候,会遇到torch版本不一致问题,尤其是torch1.1.0一直都在安装错误,试了很多方法都没用。

13610
领券