AI Benchmark Alpha adalah pustaka python sumber terbuka untuk mengevaluasi kinerja AI berbagai platform perangkat keras, termasuk CPU, GPU, dan TPU. Tolok ukur ini mengandalkan pustaka pembelajaran mesin TensorFlow, dan memberikan solusi yang ringan dan akurat untuk menilai inferensi dan kecepatan pelatihan untuk model-model Pembelajaran Mendalam yang utama.
Secara total, AI Benchmark terdiri dari 42 tes dan 19 bagian yang disediakan di bawah ini:
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]
Untuk informasi dan hasil lebih lanjut, silakan kunjungi situs web proyek: http://ai-benchmark.com/alpha
sudo docker build -t odyssee/ai-benchmark:latest"
sudo docker run --gpus all -it odyssee/ai-benchmark:latest"