| #!/bin/bash | |
| export IBA_CONFIG=./config_draccus/config.yaml | |
| export TOKENIZERS_PARALLELISM=true | |
| # CUDA Include (/cuda.h) | |
| CUDA_INCLUDE_PATH="/home/work/miniconda3/envs/allm/include" | |
| export CPATH=$CPATH:$CUDA_INCLUDE_PATH | |
| export CPLUS_INCLUDE_PATH=$CPLUS_INCLUDE_PATH:$CUDA_INCLUDE_PATH | |
| export WANDB_PROJECT="IBA_CMS" | |
| date +"%F %T" | |
| accelerate launch --dynamo_backend no --main_process_port 41353 -m src.cms_eval \ | |
| --config_path $IBA_CONFIG --infer.model_path exps/CMS/t=30d23h18m55,mlr1.0e-04,b16,r64,n_ct8,t30d23h18m55,initdef,dr0.0,ep3.0,ds14607 |