2026年7月14日火曜日

CPUのみでAIの学習をするmini-yolov5を作ってみました。

自動微分やLLMの学習の高速化の勉強をしてたら思ったのですが、AIの学習ってCPUのみでも行けるんじゃない?

また自分でも作れると思ってしまった。

CPUのみでAIの学習をするシリーズを作ってみようと思い作ってみました。


第一回はまだ誰も作ってないだろうmini-yolov5です。

yoloもレイヤーを減らして、画像を128x128にすればcpuでも数分で学習できます。

yoloの仕組みを学習するのに最適



https://github.com/yomei-o/mini-yolov5-cpp



0 件のコメント:

コメントを投稿