2019年8月11日
Pytorch迁移学习
摘要： 环境： Pytorch1.1，Python3.6，win10/ubuntu18，GPU 正文 项目结构如下所示 pokemon里面存放数据，分别是五个文件夹，其中每个文件夹分别存放一定数量的图片，总共1000多张图片； best.mdl是保存下来的模型，可以直接加载进行分类 resnet.py是自己 阅读全文
posted @ 2019-08-11 00:37 嶙羽 阅读(513) 评论(0) 推荐(0) 编辑