在 PyTorch 中，`pin_memory` 技术用于锁定页面内存。当在创建 `DataLoader` 时将 `pin_memory` 参数设置为 `True`，这意味着生成的 Tensor 数据最初会被存储在锁定的内存中。这一技术能够显著提高数据从 CPU 到 GPU 的传输效率，从而加快训练速度。通过合理利用 `pin_memory`，可以有效减少数据加载的瓶颈，提升整体性能。 ... [详细]

乾明发自凹非寺量子位报道|公众号QbitAI道友留步！TensorFlow已重获新生。在“PyTorch真香”的潮流中，有人站出来为TensorFlow说话了。这次来自谷歌的工程师 ... [详细]

1.如何进行迁移 使用Pytorch写的模型： 对模型和相应的数据使用.cuda()处理。通过这种方式，我们就可以将内存中的数据复制到GPU的显存中去。 ... [详细]

本文内容是根据莫烦Python网站的视频整理的笔记，笔记中对代码的注释更加清晰明了,同时根据所有笔记还整理了精简版的思维导图,可在此专栏查看,想观看视频可直接去他的网 ... [详细]

目录1、注册算子2、算子分发3、修改opmethods基类4、下发算子5、添加wrapper6、添加wrapper7、算子测试本教程分享了在寒武纪设备上pytorch-mlu中添加 ... [详细]

作者主页(文火冰糖的硅基工坊)：https:blog.csdn.netHiWangWenBing本文网址：https:blog.csdn.netHiW ... [详细]

上一篇目录标题网络模型训练小插曲训练模型数据训练GPU训练第一种方式方式二：查看GPU信息完整模型验证网络模型训练小插曲区别importtorchatorch ... [详细]

一、Transorboard使用（可视化工具）（观察模型不同阶段的数据状况）fromtorch.utils.tensorboardimportSummaryWriterfromPI ... [详细]

CBAM:ConvolutionalBlockAttentionModule论文地址：https:arxiv.orgabs1807.06521简介：我们提出了 ... [详细]

1.首先在终端中输入python进入python交互式环境2.接着输入 importtorch print(torch.__version__)#注意是 ... [详细]

上期使用LSTM做短文本分类效果不如CNN，或许有朋友会问什么场景下LSTM能体现出序列性的优势，本期使用双向LSTM-CNNs-CRF实现SequenceLabeling。CRF ... [详细]

本文主要是针对人脸识别中的各种loss进行总结。 背景对于分类问题，我们常用的lossfunction是softmax，表示为：,当然有softmax肯定也有hardmax:，so ... [详细]