AI Benchmark Alpha est une bibliothèque Python open source permettant d'évaluer les performances de l'IA de diverses plates-formes matérielles, notamment les CPU, les GPU et les TPU. Le benchmark s'appuie sur la bibliothèque d'apprentissage automatique TensorFlow et fournit une solution légère et précise pour évaluer la vitesse d'inférence et de formation pour les modèles clés de Deep Learning.
Au total, AI Benchmark comprend 42 tests et 19 sections présentées ci-dessous :
MobileNet-V2 [classification]
Création-V3 [classification]
Création-V4 [classification]
Création-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]
Pour plus d'informations et de résultats, veuillez visiter le site Web du projet : http://ai-benchmark.com/alpha
sudo docker build -t odyssee/ai-benchmark:latest"
sudo docker run --gpus all -it odyssee/ai-benchmark:latest"