- 博客(2)
- 收藏
- 关注
RSS订阅原创 detach 函数理解笔记
detach函数表示将tensor变量的梯度删除，从而实现减少内存的目的 举个简单的例子： # 自动求梯度 import torch x = torch.tensor(1.,requires_grad = True) w = torch.tensor(2.,requires_grad = True) b = torch.tensor(3.,requires_grad = True) y = w*x+b; z = y**2; z.backward(); print('w.grad:',w.grad,'
2021-10-18 09:11:42
518
原创 DataLoader 基础学习笔记
在自学词向量时，遇到了DataLoader这个知识点，最开始不太理解，第二次学习的时候渐渐明白了它的作用。记录一下，给相似的小白学习用。 1、DataLoader的作用 DataLoader 的作用是将数据集（Dataset）自动形成一个一个的Batch，以用于批处理使用。建立DataLoader需要先建立数据集。 相关包的导入 import torch import torch.utils.data as tud; 2、DataSet的建立 2.1 直接采用 torch.utils.data
2021-10-16 09:59:21
367
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人