| export PATH=/root/miniconda3/bin:$PATH | |
| export HF_ENDPOINT=https://hf-mirror.com | |
| export PYTHONPATH=/root/autodl-tmp/covt_repro/CoVT/train:/root/autodl-tmp/covt_repro/CoVT/train/src:/root/autodl-tmp/covt_repro/CoVT/train/src/training:${PYTHONPATH:-} | |
| export PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True | |
| BASE=/root/autodl-tmp/covt_repro | |
| COVT_CODE=$BASE/CoVT | |
| DATA_DIR=$BASE/covt_data | |
| SCRIPTS_DIR=$BASE/scripts | |
| DS_CONFIG=$SCRIPTS_DIR/zero2_pro6000.json | |
| IMAGE_MIN_PIXELS=$((256*28*28)) | |
| IMAGE_MAX_PIXELS=$((1024*28*28)) | |
| IMAGE_RESIZE_W=448 | |
| IMAGE_RESIZE_H=448 | |
| PER_DEV_BS=1 | |
| GRAD_ACCUM=8 | |