DeepLearningExamples/PyTorch/Classification/RN50v1.5/examples/RN50_FP16_INFERENCE_BENCHMARK.sh
2018-10-01 22:51:24 +02:00

4 lines
178 B
Bash

# This script launches ResNet50 inference benchmark in FP16 on 1 GPU with 256 batch size
python ./main.py -j5 --arch resnet50 -b 256 --fp16 --benchmark-inference /data/imagenet