自動微分やLLMの学習の高速化の勉強をしてたら思ったのですが、AIの学習ってCPUのみでも行けるんじゃない?
また自分でも作れると思ってしまった。
CPUのみでAIの学習をするシリーズを作ってみようと思い作ってみました。
第一回はまだ誰も作ってないだろうmini-yolov5です。
yoloもレイヤーを減らして、画像を128x128にすればcpuでも数分で学習できます。
yoloの仕組みを学習するのに最適
https://github.com/yomei-o/mini-yolov5-cpp
0 件のコメント:
コメントを投稿