PyTorch Ignite 0.4.8 : Tutorials : Getting Started on Cloud TPU (翻訳/解説)

【PyTorch Ignite 0.4.8 : Tutorials : Getting Started on Cloud TPU】
Ignite 0.4.8 をベースにドキュメントを幾つか翻訳しています。今回はチュートリアルから「Getting Started on Cloud TPU」です。

このチュートリアルでは、まず Colab 上に PyTorch/XLA をインストールして TPU で PyTorch を利用できるようにします。PyTorch/XLA は PyTorch を Cloud TPU に接続させて TPU コアをデバイスとして使用できるパッケージです。そして PyTorch-Ignite で MNIST 上で基本的なモデルを訓練します。

Ignite は PyTorch の公式エコシステムの一つで、PyTorch でニューラルネットワークを訓練するための高位ライブラリです。それは簡潔でありながらフル機能の訓練/評価ループを透過的に書くのに役立ちます。 [More]