ai benchmark

AI Benchmark Alpha é uma biblioteca python de código aberto para avaliar o desempenho de IA de várias plataformas de hardware, incluindo CPUs, GPUs e TPUs. O benchmark depende da biblioteca de aprendizado de máquina TensorFlow e fornece uma solução leve e precisa para avaliar inferência e velocidade de treinamento para os principais modelos de Deep Learning.

No total, o AI Benchmark consiste em 42 testes e 19 seções fornecidas abaixo:

1. MobileNet-V2 [classification]

2. Inception-V3 [classification]

3. Inception-V4 [classification]

4. Inception-ResNet-V2 [classification]

5. ResNet-V2-50 [classification]

6. ResNet-V2-152 [classification]

7. VGG-16 [classification]

8. SRCNN 9-5-5 [image-to-image mapping]

9. VGG-19 [image-to-image mapping]

10. ResNet-SRGAN [image-to-image mapping]

11. ResNet-DPED [image-to-image mapping]

12. U-Net [image-to-image mapping]

13. Nvidia-SPADE [image-to-image mapping]

14. ICNet [image segmentation]

15. PSPNet [image segmentation]

16. DeepLab [image segmentation]

17. Pixel-RNN [inpainting]

18. LSTM [sentence sentiment analysis]

19. GNMT [text translation]

Para mais informações e resultados, visite o site do projeto: http://ai-benchmark.com/alpha

Instruções de execução

sudo docker build -t odyssee/ai-benchmark:latest" sudo docker run --gpus all -it odyssee/ai-benchmark:latest"