- 博客(64)
- 收藏
- 关注
原创 图像数据与显存
本文介绍了图像数据处理和神经网络模型定义的相关内容,主要包括: 图像数据预处理:通过torchvision.transforms进行归一化和标准化处理,并展示了MNIST和CIFAR-10数据集的加载与可视化方法。 神经网络模型定义:详细说明了处理黑白和彩色图像的两层MLP网络结构,包括输入展平、全连接层设置和参数计算。 显存占用分析:探讨了模型参数与梯度、优化器状态、批量数据大小对显存的影响,给出了显存占用计算方法,并比较了不同batch size对训练效果的影响。 优化器选择:对比了SGD和Adam优化
2025-08-13 10:27:31
346
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人
RSS订阅
1