File size: 554 Bytes
a0d95b0 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
#!/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 |