Spaces:
Running
on
Zero
Running
on
Zero
File size: 363 Bytes
fb6a167 |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
# Specify the config file path and the GPU devices to use
# export CUDA_VISIBLE_DEVICES=0,1
# Specify the config file path
export XFL_CONFIG=./train/config/subject_512.yaml
# Specify the WANDB API key
# export WANDB_API_KEY='YOUR_WANDB_API_KEY'
echo $XFL_CONFIG
export TOKENIZERS_PARALLELISM=true
accelerate launch --main_process_port 41353 -m src.train.train |