PyTorch CNN 训练/测试/绘图 模块

在学习到 d2l 中的 CNN 板块时，发现对于大部分 CNN，除了架构网络的代码不同以外，其训练和测试的代码大部分情况都是相同的，为了简化代码的书写特地编写了这个模块（雾）。 该模块基于 PyTorch 实现。 如何使用该模块请看我的这篇文章：https://blog.csdn.net/raelum/article/details/124716176 该模块支持 CNN 的训练和测试，同时可绘制训练集/测试集的损失函数曲线，训练集/测试集的精度。此外，该模块能够计算 GPU 的读取速度。