docs: Add multi-GPU usage to module docstring of finetune_asr_model

alexandrainst · Sep 26, 2024 · 9cb943f · 9cb943f
1 parent e31c03b
commit 9cb943f
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/src/scripts/finetune_asr_model.py b/src/scripts/finetune_asr_model.py
@@ -1,7 +1,15 @@
 """Finetune a speech recognition model.
 
 Usage:
-    python src/scripts/finetune_asr_model.py [key=value] [key=value] ...
+    Running on a single GPU:
+
+    python src/scripts/finetune_asr_model.py \
+        [key=value] [key=value] ...
+
+    Running on multiple GPUs:
+
+    accelerate launch [--use-deepspeed] src/scripts/finetune_asr_model.py \
+        [key=value] [key=value] ...
 """
 
 import logging