AI Benchmark Alpha — это библиотека Python с открытым исходным кодом для оценки производительности искусственного интеллекта на различных аппаратных платформах, включая процессоры, графические процессоры и TPU. Тест основан на библиотеке машинного обучения TensorFlow и предоставляет легкое и точное решение для оценки скорости вывода и обучения для ключевых моделей глубокого обучения.
Всего AI Benchmark состоит из 42 тестов и 19 разделов, представленных ниже:
MobileNet-V2 [classification]
Начало-V3 [classification]
Начало-V4 [classification]
Inception-ResNet-V2 [classification]
ResNet-V2-50 [classification]
ResNet-V2-152 [classification]
ВГГ-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]
Пиксель-RNN [inpainting]
LSTM [sentence sentiment analysis]
ГНМТ [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"