Pytorch CIFAR10图像分类 工具函数utils更新v2.0篇

Pytorch CIFAR10图像分类 工具函数utils更新v2.0篇

文章目录

  • Pytorch CIFAR10图像分类 工具函数utils更新v2.0篇
    • 设置随机种子(保证可复现性)
    • 得到分类的精度(适配高版本)
    • 训练函数最新版本(增加tensorboard函数和tqdm可视化进度条)
    • 可视化准确率、损失、学习率变化

这里贴一下汇总篇: 汇总篇

对于上一版的工具函数utils.py,我认为可能来说,可视化的感觉还是不是很好,所以我就修改了一下我新的训练函数,为了兼容,参数基本相同,但是加入了tqdm来可视化进度条,这样也会更加的好看和直观,并且统一了一些代码的格式,使得代码稍微好看点,之前有时候有点乱,除此为了兼容一些情况,修改了部分代码,但是意义相同。

如果想看上一版的工具函数utils.py,可以查看这篇博客。PytorchCIFAR10图像分类

你可能感兴趣的:(CIFAR10图像分类,pytorch,分类,深度学习)