return mymodel
myModel相当于是全局的
数据加载器dataset
初始化函数不一定要写
一次一个 batch_size = 16一批一批丢进模型
main
定义dataset有什么用
配合dataloader,数据自动批量产生
main()
dataloader放进模型训练
重写gettraindata
return mymodel
myModel相当于是全局的
数据加载器dataset
初始化函数不一定要写
一次一个 batch_size = 16一批一批丢进模型
main
定义dataset有什么用
配合dataloader,数据自动批量产生
main()
dataloader放进模型训练
重写gettraindata