深度学习
深度学习
_Lyang_
Focusing...
展开
-
一条命令解决安装torch_scatter torch_sparse torch_cluster torch_spline
该命令安装了pyg_lib包以及一些与PyTorch相关的包(torch_scatter、torch_sparse、torch_cluster、torch_spline_conv)从特定的包索引URL。指定的PyTorch版本是2.0.0,使用的CUDA版本是11.8。如果你的版本不是我这样子的,可以输入以下指令查看自己 torch 和 cuda 的版本,然后修改上面那条命令后面的版本数字即可。原创 2024-01-14 11:00:07 · 876 阅读 · 0 评论 -
PyTorch中各种积的运算总结
原创 2024-01-05 14:40:51 · 445 阅读 · 0 评论 -
PyTorch中各种求和运算
沿某个轴计算A元素的累积总和,此函数不会沿任何轴降低输入张量的维度。降维求和会沿指定轴降低张量的维度,使它变为一个标量。原创 2023-12-28 17:28:04 · 889 阅读 · 0 评论 -
模型训练前的基本性验证(避免训练好久之后才发现模型不能用)
在模型训练前,为了防止出现训练了两天两夜才发现这个模型根本不能用的逆天问题,我们要在训练前对模型进行一个基本的验证,即Sanity Check。如果这种情况下loss都不降低,模型都不过拟合的话,那么代码一定会有问题,需要调试修改。在dataloader定义后。原创 2023-12-28 16:10:17 · 560 阅读 · 0 评论 -
batch,iteration,epoch之间的关系
原创 2023-12-28 16:01:27 · 436 阅读 · 0 评论 -
PyTorch计算损失的代码流程
【代码】PyTorch计算损失的代码流程。原创 2023-12-28 15:32:44 · 364 阅读 · 0 评论 -
深度学习需要的组件
scheduler 学习率调整器。loss_func 损失函数。optimizer 优化器。evaluator 评估器。原创 2023-12-28 15:29:09 · 407 阅读 · 0 评论 -
pandas 的 dataframe如何转 torch 的 tensor
pandas 的 dataframe -> numpy 的 array -> torch 的 tensor。pandas 的 dataframe -> torch 的 tensor。原创 2023-12-28 11:49:51 · 702 阅读 · 0 评论