Upload root_scripts/launch_v3.sh with huggingface_hub
Browse files
root_scripts/launch_v3.sh
ADDED
|
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
#!/bin/bash
|
| 2 |
+
cd /workspace/rl4phyx/RL4Phyx/SFT
|
| 3 |
+
export CUDA_VISIBLE_DEVICES=2,3,4,6
|
| 4 |
+
export PYTHONUNBUFFERED=1
|
| 5 |
+
TIMESTAMP=$(date +%Y%m%d_%H%M%S)
|
| 6 |
+
LOG="/workspace/rl4phyx/logs/train_lora_phyx_math_f_v3_${TIMESTAMP}.log"
|
| 7 |
+
echo "Starting lora_phyx_math_f (GPUs 2,3,4,6, grad_accum=16, batch=64, save_only_model=False) at $(date)" > "$LOG"
|
| 8 |
+
torchrun --nproc_per_node=4 --master_port=29500 train_sft_phyx_math_lora_freeze.py >> "$LOG" 2>&1
|
| 9 |
+
echo "TRAINING_COMPLETE at $(date)" >> "$LOG"
|