Skip to content

Commit c89c9db

Browse files
authored
fix(convert ckpt): fix option (#437)
1 parent 47d0f54 commit c89c9db

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tools/convert_ckpt_parallel.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
python tools/convert_ckpt_parallel.py \
44
<origin_ckpt_path> <target_ckpt_path> \
55
--origin_meta_path <origin_meta_path> --target_meta_path <target_meta_path> \
6-
--copy_file <True/False> --convert_optimizer <True/False>
6+
--copy_file --convert_optimizer
77
88
When meta_path is not specified, it will automatically search and load meta in the ckpt path.
99
Default to convert optimizer state and copy files.
@@ -31,8 +31,8 @@ def parse_args():
3131
args.add_argument("target_ckpt_path", type=str, default=None)
3232
args.add_argument("--origin_meta_path", type=str, default=None)
3333
args.add_argument("--target_meta_path", type=str, default=None)
34-
args.add_argument("--copy_file", type=bool, default=True, help="enable/disable copy other file.")
35-
args.add_argument("--convert_optimizer", type=bool, default=True, help="enable/disable optimizer converting.")
34+
args.add_argument("--copy_file", default=False, action="store_true", help="enable copy other file.")
35+
args.add_argument("--convert_optimizer", default=False, action="store_true", help="enable optimizer converting.")
3636
return args.parse_args()
3737

3838

0 commit comments

Comments
 (0)