AI Benchmark Alpha es una biblioteca de Python de código abierto para evaluar el rendimiento de la IA de varias plataformas de hardware, incluidas CPU, GPU y TPU. El punto de referencia se basa en la biblioteca de aprendizaje automático TensorFlow y proporciona una solución ligera y precisa para evaluar la inferencia y la velocidad de entrenamiento para modelos clave de aprendizaje profundo.
En total, AI Benchmark consta de 42 pruebas y 19 secciones que se detallan a continuación:
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]
Para obtener más información y resultados, visite el sitio web del proyecto: http://ai-benchmark.com/alpha
sudo docker build -t odyssee/ai-benchmark:latest"
sudo docker run --gpus all -it odyssee/ai-benchmark:latest"