【go语言】go语言实现go-torch， 完成Lenet-5的搭建，训练，以及pth和onnx模型导出

文章目录

• 介绍
• • 基准
  • [pytorch耗时 12秒](#pytorch耗时 12秒)
  • [go-torch耗时： 50秒](#go-torch耗时： 50秒)
  • [go-torch + openblas 37秒](#go-torch + openblas 37秒)
  • [go-torch + im2col并行优化: 15s](#go-torch + im2col并行优化: 15s)
  • [+ im2col 复用 + gradCol 池化 + bias 并行 13.5s](#+ im2col 复用 + gradCol 池化 + bias 并行 13.5s)

介绍

go版本的pytorch雏形

基准

• cpu：i7-11800h
• mnist分类器任务
• 两轮：2epoch

pytorch耗时 12秒

go-torch耗时： 50秒

go-torch + openblas 37秒

go-torch + im2col并行优化: 15s

+ im2col 复用 + gradCol 池化 + bias 并行 13.5s