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]
자세한 내용과 결과를 보려면 프로젝트 웹사이트(http://ai-benchmark.com/alpha)를 방문하세요.
sudo docker build -t odyssee/ai-benchmark:latest"
sudo docker run --gpus all -it odyssee/ai-benchmark:latest"