2024年4月15日
Pytorch分类模型的训练框架
摘要： Pytorch分类模型的训练框架 PhotoDataset数据集是自己定义的数据集，数据集存放方式为： image文件夹 0文件夹 img1.jpg img2.jpg 1文件夹 img1.jpg img2.jpg .... 如果是cpu训练的话，就把代码中的.cuda()改成.cpu() impor 阅读全文
posted @ 2024-04-15 16:09 梁君牧 阅读(54) 评论(0) 推荐(0) 编辑