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

时间:2022-05-15 12:11:16
【文件属性】:

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


网友评论