跳至主要內容
果冻の博客
博客主页
时间轴
V2 文档
open in new window
工具
boom
open in new window
搜索文档
K
人工不智能学习笔记
Jelly
小于 1 分钟
一些学习AI的心得体会
昨天跟着pytorch的入门教程操作了一遍, 对深度学习有了一定的了解.
对于入门教程中的CNN网络, 基本涉及以下操作
设定数据集
定义模型
定义训练函数和测试函数
遍历指定ephos训练函数和测试函数
保存训练好的模型
调用模型进行预测
下一页
torch.nn.Flatten 和 torch.flatten 的区别