【文件属性】：
文件名称：PyTorch CNN 训练/测试/绘图 模块
文件大小：3KB
文件格式：PY
更新时间：2022-05-15 12:11:16
cnn 源码软件 人工智能 神经网络 深度学习
在学习到 d2l 中的 CNN 板块时，发现对于大部分 CNN，除了架构网络的代码不同以外，其训练和测试的代码大部分情况都是相同的，为了简化代码的书写特地编写了这个模块（雾）。
该模块基于 PyTorch 实现。
如何使用该模块请看我的这篇文章：https://blog.csdn.net/raelum/article/details/124716176
该模块支持 CNN 的训练和测试，同时可绘制训练集/测试集的损失函数曲线，训练集/测试集的精度。此外，该模块能够计算 GPU 的读取速度。