Pytorch20天训练小伙伴，day2图片数据建模流程如何范例？

本文共计2484个文字，预计阅读时间需要10分钟。

系统教程20天掌握Pytorch+最近和中巴、会巴进行一个小打卡活动，20天Pytorch，这是第二天。欢迎一键三连。

文章目录：

一、准备数据

二、定义模型

三、训练模型

系统教程20天拿下Pytorch​ 最近和中哥、会哥进行一个小打卡活动，20天pytorch，这是第二天。欢迎一键三连。

文章目录

• ​​一、准备数据​​
• ​​二、定义模型​​
• ​​三、训练模型​​
• ​​四、评估模型​​
• ​​五、使用模型​​
• ​​六、保存模型​​
• ​​总结​​

import os
import datetime

#打印时间
def printbar():
nowtime = datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')
print("\n"+"=========="*8 + "%s"%nowtime)

#mac系统上pytorch和matplotlib在jupyter中同时跑需要更改环境变量
os.environ["KMP_DUPLICATE_LIB_OK"]="TRUE"!pip install prettytable
!pip install torchkeras

一、准备数据

cifar2数据集为cifar10数据集的子集，只包括前两种类别airplane和automobile。