文件名称: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 的读取速度。