2022-04-27から1日間の記事一覧

雑記: Flax でのモデル生成と適用

参考文献 Flax 2 ("Linen") - Colaboratory(2022年4月27日参照). JAX/Flaxを使ってMNISTを学習させてみる | TC3株式会社(2022年4月27日参照). Flax でニューラルモデルにデータを流すのは以下のようにやることになるということです。 JAX では乱数を生…