chen459664's picture
Add files using upload-large-folder tool
2051791 verified
Raw
History Blame Contribute Delete
329 Bytes
#!/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