上传者: raelum
|
上传时间: 2022-05-15 16:06:37
|
文件大小: 3KB
|
文件类型: PY
在学习到 d2l 中的 CNN 板块时,发现对于大部分 CNN,除了架构网络的代码不同以外,其训练和测试的代码大部分情况都是相同的,为了简化代码的书写特地开发了这个模块(雾)。
该模块基于 PyTorch 实现。
如何使用该模块请看我的这篇文章:https://blog.csdn.net/raelum/article/details/124716176
该模块支持 CNN 的训练和测试,同时可绘制训练集/测试集的损失函数曲线,训练集/测试集的精度。此外,该模块能够计算 GPU 的读取速度。