File size: 329 Bytes
2051791
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
#!/bin/bash

# export HF_HOME=/data/hugginface
# conda activate quant-eval

if [ ! -d logs ]; then
    mkdir logs
fi

export HF_HOME=/home/justin/.cache/huggingface/
export PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True
python ../src/lm_quant_toolkit/eval/bench_vit.py 2>&1 \
    | tee logs/bench-vit-$(date +%Y%m%d%H%M%S).log