train
2022-05-05 12:03:37 19 举报
HAHA
作者其他创作
大纲/内容
在visdom上展示图片并保存
训练model最优化参数
设置训练选项
Y
N
training flag设定,如epoch,total_iteration
显示epoch
超过最大迭代次数
每次save_iter_freq迭代,将模型保存在磁盘上
创建一个model
keep_training= true
创建dataset并显式数据集规模
每次save_latest_freq迭代,将最新模型保存在磁盘上
更新学习率
记录iteration次数
显式训练误差,并将日志信息保存在磁盘上
打印第n轮循环结束
i,data in enumerate(dataset)
创建visualizer
开始
结束
将data读入model
0 条评论
回复 删除
下一页