github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-07 03:52:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

[bug] fix get_default_parser in examples (#4764)

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

6
examples/images/vit/args.py
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,
@@ -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",