深度学习
文章平均质量分 64
Lntano__y
在这里记录自己的学习历程,并分享给有需要的同学!
展开
-
from torch.utils.data import Dataset详解
torch.utils.data.Dataset 是 PyTorch 数据加载库中的一个重要类,用于定义自定义数据集。通过继承 Dataset 类,可以创建自己的数据集类,并实现数据的加载和处理逻辑。通过继承和实现 torch.utils.data.Dataset 类,可以灵活地创建自定义数据集,并与 DataLoader 结合使用,实现高效的数据加载和处理。(self, idx):支持索引操作,返回指定索引的样本。继承 Dataset 类,并实现。方法中加载文件数据。2.1 导入相关模块。原创 2024-07-19 16:38:23 · 466 阅读 · 0 评论 -
torch.no_grad()详解
torch.no_grad() 是一个用于禁用梯度计算的上下文管理器,适用于模型评估、推理等不需要梯度计算的场景。在使用 torch.no_grad() 时,通常还会将模型设置为评估模式(model.eval()),以确保某些层(如 dropout 和 batch normalization)在推理时的行为与训练时不同。torch.no_grad() 可以嵌套使用,内层的 torch.no_grad() 仍然会禁用梯度计算。进入 torch.no_grad() 上下文,临时禁用梯度计算。原创 2024-07-19 10:46:01 · 1045 阅读 · 0 评论