--- a +++ b/scripts/benchmark.sh @@ -0,0 +1,21 @@ +#!/bin/bash + +set -e + +export CUDA_VISIBLE_DEVICES=0 +# Smaller batch size so that sequences with Flair embeddings fit in GPU memory. +python -m scripts.benchmark benchmark_gpu \ + --bilstmcrf_large_batch_size 64 \ + --bilstmcrf_fast_batch_size 64 + +export CUDA_VISIBLE_DEVICES="" +export MKL_NUM_THREADS=32 +python -m scripts.benchmark benchmark_cpu_32_threads + +export CUDA_VISIBLE_DEVICES="" +export MKL_NUM_THREADS=16 +python -m scripts.benchmark benchmark_cpu_16_threads + +export CUDA_VISIBLE_DEVICES="" +export MKL_NUM_THREADS=8 +python -m scripts.benchmark benchmark_cpu_8_threads