LambdaColdStorage / lambda-tensorflow-benchmark

BSD 3-Clause "New" or "Revised" License
234 stars 60 forks source link

Rework model/batch size configuration again #14

Open jeremybobbin opened 4 years ago

jeremybobbin commented 4 years ago
  resnet50='5  + 1/3'
 resnet152='2  + 2/3'
inception3='5  + 1/3'
inception4='1  + 1/3'
     vgg16='5  + 1/3'
   alexnet='42 + 2/3'
    ssd300='2  + 2/3'
./benchmark.sh -l 2 -h 4     # low/high - benchmarks 2 GPUs, 3 GPUs and 4GPUs at a time.

Options:

i - GPU_INDEX
l - MIN_NUM_GPU
h - MAX_NUM_GPU
n - ITERATIONS
b - NUM_BATCHES
c - SETTING
v - GPU_VENDOR
t - THERMAL_INTERVAL

This:

./batch_benchmark.sh 1 1 1 100 2 config_resnet50_replicated_fp32_train_syn

Is now this:

./benchmark.sh -h 1 -n 1 -b 100 -t 2 -c config_resnet50_replicated_fp32_train_syn
jeremybobbin commented 4 years ago

Issues: