[bug] fix get_default_parser in examples (#4764)

examples/images/vit/args.py

@@ -1,8 +1,8 @@
-from colossalai import get_default_parser
+import argparse
 
 
 def parse_demo_args():
-    parser = get_default_parser()
+    parser = argparse.ArgumentParser()
     parser.add_argument(
         "--model_name_or_path",
         type=str,
@@ -52,7 +52,7 @@ def parse_demo_args():
 
 
 def parse_benchmark_args():
-    parser = get_default_parser()
+    parser = argparse.ArgumentParser()
 
     parser.add_argument(
         "--model_name_or_path",

examples/images/vit/run_benchmark.sh

@@ -11,9 +11,9 @@ for PLUGIN in "torch_ddp" "torch_ddp_fp16" "low_level_zero" "gemini" "hybrid_par
 do
 
 MODEL_PATH="google/vit-base-patch16-224"
-torchrun \
-  --standalone \
-  --nproc_per_node 4 \
+colossalai run \
+  --nproc_per_node ${GPUNUM} \
+  --master_port 29505 \
   vit_benchmark.py \
   --model_name_or_path ${MODEL_PATH} \
   --mem_cap ${MEMCAP} \

examples/images/vit/run_demo.sh

@@ -35,9 +35,9 @@ WEIGHT_DECAY=0.05
 WARMUP_RATIO=0.3
 
 # run the script for demo
-torchrun \
-  --standalone \
+colossalai run \
   --nproc_per_node ${GPUNUM} \
+  --master_port 29505 \
   vit_train_demo.py \
   --model_name_or_path ${MODEL} \
   --output_path ${OUTPUT_PATH} \

examples/images/vit/test_ci.sh

@@ -5,9 +5,9 @@ BS=8
 for PLUGIN in "torch_ddp" "torch_ddp_fp16" "low_level_zero" "gemini" "hybrid_parallel"
 do
 
-torchrun \
-  --standalone \
+colossalai run \
   --nproc_per_node 4 \
+  --master_port 29505 \
   vit_benchmark.py \
   --model_name_or_path "google/vit-base-patch16-224" \
   --plugin ${PLUGIN} \