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

5 lines
249 B
Bash

# This script evaluates ResNet50 model in FP16 using 64 batch size on 1 GPU
# Usage: ./RN50_FP16_EVAL.sh <path to this repository> <path to checkpoint>
python $1/main.py -j5 p 100 --arch resnet50 -b 256 --resume $2 --evaluate --fp16 /data/imagenet