AI Benchmark Alpha ist eine Open-Source-Python-Bibliothek zur Bewertung der KI-Leistung verschiedener Hardwareplattformen, einschließlich CPUs, GPUs und TPUs. Der Benchmark basiert auf der TensorFlow-Bibliothek für maschinelles Lernen und bietet eine einfache und genaue Lösung zur Bewertung der Inferenz- und Trainingsgeschwindigkeit für wichtige Deep-Learning-Modelle.
Insgesamt besteht AI Benchmark aus 42 Tests und den folgenden 19 Abschnitten :
MobileNet-V2 [classification]
Inception-V3 [classification]
Inception-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]
Weitere Informationen und Ergebnisse finden Sie auf der Projektwebsite: http://ai-benchmark.com/alpha
sudo docker build -t odyssee/ai-benchmark:latest"
sudo docker run --gpus all -it odyssee/ai-benchmark:latest"