RISC-VプロセッサHiFive1でMNISTを動作させよう。実機デバッグをしていて、やっと動作するようになった。 まだバッチサイズは1のままだが、とりあえず動作するようになった。
問題だったのは、データの配置方法だった。ちゃんと調整して入力データを流すと、動作するようになった。 下記の通り、入力画像データに対して、ちゃんと数値を認識できている。
とりあえず動作するようになったので、次の課題としては、
- プログラムの高速化、軽量化
- バッチサイズを増やす
など。HiFive1ならでは、っていうはなかなかか難しそうだけど、出来るところまでやってみようかな。