PyTorch
关注
分享
扫一扫
William.csj
华中科技大学在读博士生。
研究方向：多目标跟踪，多模态感知。
硕博经历：获得国家奖学金等多项奖学金，以第一作者发表两篇人工智能国际顶级会议CCF A类论文（CVPR、AAAI），以非第一作者发表的论文（ICCV、ICLR）。
本科经历：获得国家奖学金等十余项奖学金，参加数学竞赛、数学建模竞赛、ROBOCON机器人比赛、互联网+、挑战杯等，曾获数学竞赛全国一、二等奖、数学建模竞赛多次国家（际）一、二等奖、ROBOCON机器人两次全国二等奖（队长）、互联网+全国铜奖（省最佳创意奖）等十余项国家级奖项，曾在深圳科创学院-智能驾驶中心担任视觉感知算法实习生。
展开
专栏收录文章
- 默认排序
- 最新发布
- 最早发布
- 最多阅读
- 最少阅读
-
服务器/Pytorch——对于只调用一次的函数初始化，放在for训练外面和里面的差异
服务器/Pytorch——对于只调用一次的函数初始化，放在for训练外面和里面的差异原创 2025-10-20 14:46:19 · 380 阅读 · 0 评论 -
Pytorch/CUDA——flash-attn 库编译的 gcc 版本问题
flash-attn编译依赖特定版本的gcc和CUDA Toolkit组合。不同CUDA版本对应支持的gcc范围不同（如CUDA 11.8需gcc 9.x~11.x），版本不匹配会导致nvcc报错。需确保系统gcc、nvcc和PyTorch的CUDA版本三者兼容。可通过命令查看当前版本，若版本不匹配需调整gcc版本。flash-attn必须使用nvcc编译，不支持替代方案。原创 2025-07-04 11:06:45 · 1483 阅读 · 0 评论 -
Pytorch——查看模型的推理引擎
print(type(model))原创 2025-07-01 11:58:36 · 294 阅读 · 0 评论 -
Pytorch——pip下载安装pytorch慢的解决办法
Pytorch——pip下载安装pytorch慢的解决办法原创 2024-10-14 11:56:40 · 6173 阅读 · 2 评论 -
报错解决—— element 0 of tensors does not require grad and does not have a grad_fn
backward()的变量的requires_grad是False，会导致不会计算梯度，从而报错。原创 2024-05-18 22:25:04 · 462 阅读 · 0 评论 -
PyTorch——List转Tensor的方法
PyTorch——List转Tensor的方法原创 2024-05-17 20:01:24 · 1890 阅读 · 0 评论 -
Torch——torch.Tensor.detach()解析
Torch——torch.Tensor.detach()解析转载 2024-04-25 15:07:46 · 297 阅读 · 0 评论 -
优化器——torch.optim.SGD 和 torch.optim.Adam的区别
优化器——torch.optim.SGD 和 torch.optim.Adam的区别转载 2024-04-25 11:09:19 · 283 阅读 · 0 评论 -
Pytorch——训练时，冻结网络部分参数的方法
要固定训练网络的哪几层，只需要找到这几层参数(parameter)，然后将其 `.requires_grad` 属性设置为 `False` 。然后`修改优化器`，只将不被冻结的层传入。原创 2024-04-17 10:46:31 · 3899 阅读 · 0 评论 -
报错解决——多batchsize时，RuntimeError: each element in list of batch should be of equal size
报错解决——多batchsize时，RuntimeError: each element in list of batch should be of equal size原创 2024-04-16 18:35:34 · 1212 阅读 · 0 评论 -
PyTorch——torchtext与PyTorch匹配的版本
PyTorch——torchtext与PyTorch匹配的版本。原创 2024-01-15 14:36:28 · 2857 阅读 · 0 评论 -
Pytorch——nn.LayerNorm() 和 nn.BatchNorm()函数解析
BN是对一个batch-size样本内的每个特征做归一化，LN是对每个样本的所有特征做归一化转载 2023-12-26 17:20:23 · 1315 阅读 · 0 评论 -
Pytorch——squeeze()函数
降维函数转载 2023-12-26 10:08:20 · 239 阅读 · 0 评论 -
Pytorch——unsqueeze()函数理解
升维转载 2023-12-25 13:19:42 · 678 阅读 · 0 评论 -
Pytorch——nn.BatchNorm()函数
Pytorch——nn.BatchNorm()函数转载 2023-12-22 21:40:44 · 217 阅读 · 0 评论 -
Pytorch——nn.AdaptiveAvgPool2d()函数
自适应平均池化，指定输出（H，W）转载 2023-12-22 17:55:19 · 198 阅读 · 0 评论 -
Pytorch——torch.nn.Conv2d()函数
Pytorch——torch.nn.Conv2d()函数转载 2023-12-22 15:45:03 · 181 阅读 · 0 评论 -
Pytorch——torch.arange()函数
Pytorch——torch.arange()函数转载 2023-12-22 14:48:55 · 240 阅读 · 0 评论 -
Pytorch——nn.Flatten()函数
nn.Flatten()：对图像数组进行展平操作转载 2023-12-22 10:50:48 · 931 阅读 · 0 评论 -
Pytorch——激活函数层
Pytorch——激活函数层转载 2023-12-22 09:59:48 · 280 阅读 · 0 评论 -
Pytorch——nn.Dropout()函数
Pytorch——nn.Dropout()函数转载 2023-12-22 09:57:42 · 340 阅读 · 0 评论 -
Pytorch——torch.init.normal_()函数
torch.init.normal_：给tensor初始化，一般是给网络中参数weight初始化，初始化参数值符合正态分布。转载 2023-12-21 21:47:03 · 857 阅读 · 0 评论 -
Pytorch——nn.Dropout()
nn.Dropout转载 2023-12-18 21:34:02 · 179 阅读 · 0 评论 -
Pytorch——torch.gather()
从原tensor中获取指定dim和指定index的数据转载 2023-12-13 16:58:07 · 115 阅读 · 0 评论 -
Pytorch——torch.nn.functional.pad()函数
Pytorch——torch.nn.functional.pad()函数转载 2023-12-13 16:11:32 · 321 阅读 · 0 评论 -
Pytorch——torch.cumprod() 累积乘法
Pytorch——torch.cumprod() 累积乘法转载 2023-12-13 15:40:58 · 339 阅读 · 0 评论 -
Pytorch——tensor与numpy的相互转换
Pytorch——tensor与numpy的相互转换转载 2023-09-21 11:57:43 · 376 阅读 · 0 评论 -
Pytorch——报错解决：已经安装torch，却报错找不到torch
将虚拟环境路径添加到 `.bashrc`文件内，更新 .bashrc文件。原创 2023-05-13 10:16:45 · 7394 阅读 · 0 评论 -
报错解决——RuntimeError: PytorchStreamReader failed reading zip archive: failed finding central directory
模型文件损坏，重新下载该文件原创 2023-05-12 10:28:12 · 1280 阅读 · 0 评论 -
Pytorch——关于模型调用（推理）时是否需要GPU的理解
训练时用到啥，推理的时候就会用到啥。原创 2023-04-17 16:39:43 · 1221 阅读 · 0 评论 -
Numpy——次方np.power()函数
np.power()原创 2023-02-21 14:45:07 · 684 阅读 · 0 评论 -
Pytorch——图像预处理transforms的二十二个方法
Pytorch——图像预处理transforms的二十二个方法转载 2023-02-20 21:10:50 · 277 阅读 · 0 评论 -
Tensorboard——训练可视化方法
Tensorboard——训练可视化方法原创 2023-02-15 09:45:07 · 212 阅读 · 0 评论 -
Pytorch——任意多卡GPU运行网络
Pytorch——任意多卡GPU运行网络原创 2023-02-09 10:12:46 · 358 阅读 · 0 评论 -
Pytorch——tensor转普通数字
.item()、.tolist()原创 2023-01-03 21:16:56 · 13998 阅读 · 0 评论 -
Pytorch——narrow()切片函数详解
narrow()：用于tensor切片原创 2022-12-26 20:02:03 · 1343 阅读 · 0 评论 -
报错解决——RuntimeError: PytorchStreamReader failed reading zip archive: failed finding central directory
模型调用报错的解决方案原创 2022-08-02 17:40:14 · 40308 阅读 · 8 评论 -
Pytorch——报错解决：RuntimeError: Expected all tensors to be on the same device, but found at least two
tensors 一会在CPU运算，一会在GPU运算。都改成在GPU运算。原创 2022-07-12 16:31:37 · 15076 阅读 · 1 评论 -
Pytorch环境配置——cuda、、cudnn、torch、torchvision对应版本（最全）及安装方法
Pytorch环境配置——cuda、、cudnn、torch、torchvision对应版本（最全）及安装方法原创 2022-04-02 16:05:21 · 67365 阅读 · 0 评论 -
PyTorch——报错解决：RuntimeError: CUDA out of memory. Tried to allocate 20.00 MiB (GPU 0； 3.94 GiB total c
1、完整报错RuntimeError: CUDA out of memory. Tried to allocate 20.00 MiB (GPU 0; 4.00 GiB total capacity; 2.41 GiB already allocated; 5.70 MiB free; 2.56 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_si原创 2022-01-10 17:08:42 · 18598 阅读 · 6 评论