打卡P4周:猴痘病识别题

本文记录了深度学习训练营中关于不同优化器的对比实验,涉及数据集划分、标准化处理和CNN网络构建。作者复习了torch库的DataLoader和random_split方法,以及解决了一个训练过程中的错误。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

一. 前期准备

1. 设置GPU

与前面的一致

2. 导入数据

与p3一致

输出结果

复习了一下glob()的用法.

打印classeNames列表,显示每个文件所属的类别名称。这一次相比于上一次仔细看了列表,显示了数量,以及transform.compose()下的参数。

学习到了,需要标准化处理,以达到更容易收敛的目的。

3. 划分数据集

学习到了可以直接通过len函数得到dataset创建的对象的图片总数。

本课程又介绍了一遍torch.utils.data.DataLoader()参数详解。这次根据程序推断出了输出结果是:

然后我发现了在p3忽略掉的torch.utils.data.random_split()的用法:

该方法将总体数据total_data按照指定的大小比例([train_size, test_size])随机划分为训练集和测试集,并将划分结果分别赋值给train_dataset和test_dataset两个变量。

二. 构建简单cnn网络/训练模型

此次代码复现报错,RuntimeError:省略.. The "freeze_support()" line can be omitted if the program ...

通过加 if name == 'main': 语句,解决问题。

epoch划分了20个,这里仅展示8个:

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值