- 博客(3)
- 收藏
- 关注
原创 【Pytorch代码学习】——训练部分
代码基于花卉识别训练train代码中部分函数讲解1.tqdm就是展示进度条的函数2.net.train()、net.eval()目的验证中不dropout完整训练代码net = AlexNet(num_classes=5, init_weights=True) net.to(device) loss_function = nn.CrossEntropyLoss() # pata = list(net.parameters()) optimizer = op
2022-01-21 01:02:20
1219
原创 【Pytorch代码学习】——数据集划分
简介将数据集划分为训练集和测试集代码介绍目录文件目录存放格式运行前运行后代码import osfrom shutil import copy, rmtreeimport randomdef mk_file(file_path: str): if os.path.exists(file_path): # 如果文件夹存在,则先删除原文件夹在重新创建 rmtree(file_path) os.makedirs(file_pat
2022-01-20 22:59:36
6129
3
原创 【pytorch学习笔记1】——数据集读取
数据读取Dataset与Dataloader前言图片数据集(标签在图片名称上)Datasetdata列表构建总结待续前言在pytorch学习这一块总是断断续续,完成大作业所写的代码再次回首已经完全看不懂了。所以我决定把学习过程中遇到的一些问题和知识总结出来,希望能取得一些进步吧。本人完全菜鸟,写这些笔记的主要目的是督促自己坚持学习下去,笔记中可能出现比较夸张的错误,恳请各位大佬谅解。在数据集读取学习过程中遇到了很多很多很多困难,目前也只是对图片数据集(标签信息在图片名称上)读取稍微明白了一些,关于tx
2022-01-19 21:29:27
3379
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人