- 博客(3)
- 收藏
- 关注
RSS订阅原创 VIT代码遇到的函数及作用
(self.num_attention_heads, self.attention_head_size)，是一个tuple故可以相加。其中x.size() 是一个torch.size变量 ，是tuple的子类，可以按照tuple计算，
2025-07-04 09:57:34
130
原创 pytorch训练神经网络中随机数种子的相关函数
在Pytorch中使用GPU时，底层的CuDNN库负责加速神经网络的运算。CuDNN库在每次运行前会自动选择一种算法来执行计算。torch.backends.cudnn.deterministic = True 可以保证每一次训练的算法都是相同的。在使用pytorch在gpu上训练RNN网络时涉及到了确保每一次随机数保持一致的要求，从而避免随机因素引起的神经网络效果差异，使网络可以复现。是用来保证numpy的随机数一致。是用来保证cpu的随机数一致。是用来保证gpu的随机数一致。
2025-06-30 10:13:51
173
原创 GTX1050 CUDA92 PIP 安装pythorch
摘要：在检查PyTorch历史版本时发现，官网显示V1.7.1有CU92版本，但实际下载页面仅提供Linux版。因此只能选择V1.5.1版本，且该版本最高支持Python 3.8。
2025-06-11 12:34:46
95
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人