# for training

deepspeed train.py \
  --model_name "[Model path here]" \
  --output_dir "[Output model path]" \
  --train_data_path "[Training data path here]" \
  --deepspeed_config "[Deepspeed configuration path]" \
  --master_addr "[MASTER_ADDR]" \
  --master_port "[MASTER_PORT]" \
  --num_nodes "[SLURM_NNODES]" \
  --num_gpus "[GPUS_PER_NODE]"

# for testing
python evaluation.py --model_path "[Model path here]" --data_path "[Data path here]" > evaluation.log 2> evaluation_error.log