[bug] fix get_default_parser in examples (#4764)

This commit is contained in:
Baizhou Zhang
2023-09-21 10:42:25 +08:00
committed by GitHub
parent c0a033700c
commit df66741f77
14 changed files with 36 additions and 26 deletions

View File

@@ -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,
@@ -39,7 +39,7 @@ def parse_demo_args():
def parse_benchmark_args():
parser = get_default_parser()
parser = argparse.ArgumentParser()
parser.add_argument(
"--model_name_or_path",
type=str,

View File

@@ -16,9 +16,9 @@ for GPUNUM in 1 4
do
MODLE_PATH="facebook/opt-${MODEL}"
torchrun \
--standalone \
colossalai run \
--nproc_per_node ${GPUNUM} \
--master_port 29505 \
opt_benchmark.py \
--model_name_or_path ${MODLE_PATH} \
--mem_cap ${MEMCAP} \

View File

@@ -30,9 +30,9 @@ WEIGHT_DECAY=0.01
WARMUP_RATIO=0.1
# run the script for demo
torchrun \
--standalone \
colossalai run \
--nproc_per_node ${GPUNUM} \
--master_port 29505 \
opt_train_demo.py \
--model_name_or_path ${MODEL} \
--output_path ${OUTPUT_PATH} \

View File

@@ -7,9 +7,9 @@ do
for GPUNUM in 1 4
do
torchrun \
--standalone \
colossalai run \
--nproc_per_node ${GPUNUM} \
--master_port 29505 \
opt_benchmark.py \
--model_name_or_path "facebook/opt-125m" \
--plugin ${PLUGIN} \