AI Benchmark Alpha は、CPU、GPU、TPU などのさまざまなハードウェア プラットフォームの AI パフォーマンスを評価するためのオープンソース Python ライブラリです。このベンチマークは TensorFlow 機械学習ライブラリに依存しており、主要な深層学習モデルの推論とトレーニング速度を評価するための軽量で正確なソリューションを提供します。
合計で、AI ベンチマークは、以下に示す42 のテストと19 のセクションで構成されます。
MobileNet-V2 [classification]
インセプション-V3 [classification]
インセプション-V4 [classification]
Inception-ResNet-V2 [classification]
ResNet-V2-50 [classification]
ResNet-V2-152 [classification]
VGG-16 [classification]
SRCNN 9-5-5 [image-to-image mapping]
VGG-19 [image-to-image mapping]
ResNet-SRGAN [image-to-image mapping]
ResNet-DPED [image-to-image mapping]
U-Net [image-to-image mapping]
Nvidia-SPADE [image-to-image mapping]
ICNet [image segmentation]
PSPNet [image segmentation]
DeepLab [image segmentation]
Pixel-RNN [inpainting]
LSTM [sentence sentiment analysis]
GNMT [text translation]
詳細と結果については、プロジェクトの Web サイトをご覧ください: http://ai-benchmark.com/alpha
sudo docker build -t odyssee/ai-benchmark:latest"
sudo docker run --gpus all -it odyssee/ai-benchmark:latest"