# Step 1
#   - Add dataset to dataset_info.json

cd /shared/storage-01/users/mehri2/LLaMA-Factory

# Step 2: Train the model
llamafactory-cli train /shared/storage-01/users/mehri2/LLaMA-Factory/examples/train_full/collaborative_agent/llama3_full_sft_session_level_reflection.yaml >> train_llama3.out 2>&1

llamafactory-cli train /shared/storage-01/users/mehri2/LLaMA-Factory/examples/train_full/collaborative_agent/qwen2.5_full_sft_session_level_reflection.yaml >> train_qwen2.5.out 2>&1