mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-09-08 04:24:47 +00:00
[bug] fix get_default_parser in examples (#4764)
This commit is contained in:
@@ -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,
|
||||
|
@@ -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} \
|
||||
|
@@ -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} \
|
||||
|
@@ -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} \
|
||||
|
Reference in New Issue
Block a user