- 博客(6)
- 收藏
- 关注
原创 torchvision中数据集和DataLoader的使用
目录一、torchvision中的数据集1.参数2.代码示例二、DataLoader的使用1.参数2.代码示例CIFAR10数据集root:数据集所在的位置train:train=True时,为训练集,为False时,是测试集transform:对数据集进行的操作(裁剪,归一化等)target_transform:对target进行的transformdownload:True时,自动下载数据集,False时,不下载dataset:数据集batch_size:一次所拿取的数量shuffle:为True
2025-06-18 16:40:45
170
原创 Transform
在我看来,Transform就像是一个工具箱,里面存放着 totensor, resize 等工具,将你输入的图片用这些工具进行处理,然后输出。下面来介绍一些常用工具。
2025-06-18 16:22:45
397
原创 Tensorboard
TensorBoard 是一种可视化工具,可用于监控深度学习模型的训练过程、可视化网络结构、分析参数分布等。它能将枯燥的训练数据转化为直观的图表,帮助开发者调试模型、优化训练流程。
2025-06-15 16:11:05
412
原创 pytorch环境配置(cpu版本)
访问pytorch官网,选择cpu版本,在激活的虚拟环境中,使用pip安装pytorch(这里可以安装国内源来加快安装速度)输入conda activate <env_name>激活虚拟环境。首先,访问Anaconda官网,下载安装Anaconda。1.打开Anaconda Prompt。
2025-06-14 17:04:05
231
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人