在GPU上Lenet5模型Mindspore框架实现
以下代码是Lenet模型在Mindspore框架上的简单实现,代码可以跑通,但结果还存在优化空间,可能与本实验是在GPU上跑Mindspore模型有关,同样的模型在pytorch上结果很好,但在Mindpore上性能一般。后续准备在华为服务器上用同样代码再进行一下实验。结果做了一点简单的可视化展示,绘制了loss和acc曲线以及混淆矩阵。,包含25000张猫狗图像。
原创
2024-08-12 10:19:27 ·
285 阅读 ·
0 评论