タグ

ブックマーク / note.com/hamachi_jp (1)

  • PyTorchとJAXに対応したKeras3でMNISTを試す|はまち

    バックボーンのフレームワークを、従来のTensorFlowから、デファクトスタンダードになりつつあるPyTorchと、実行効率に優れたJAXも選べるようになったKeras3.0が公開されていたので、さっそくバックボーンをPyTorchやJAXに設定して、手書きアルファベット画像のクラス分け課題のMNISTを試してみました。 23.11.29追記 公式の紹介ページも公開されていました。 https://keras.io/keras_3/ Keras3のインストール、インポート今回はGoogle Colabで試してみます。Keras3は現時点ではPyPI上では、プレビューリリースとしてkeras-coreの名前でインストールできます。 !pip install keras-coreバックエンドの設定(torch, jax, tensorflow) import os os.environ["K

    PyTorchとJAXに対応したKeras3でMNISTを試す|はまち
    sh19910711
    sh19910711 2024/05/03
    "Keras3.0: デファクトスタンダードになりつつあるPyTorchと、実行効率に優れたJAXも選べるようになった / バックエンドをJAXにするとTensorFlowと比べてざっくり2倍程度、学習および推論が高速化" 2023
  • 1