Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
Megatron-LM
Commits
5a1b96fa
Commit
5a1b96fa
authored
Jan 08, 2025
by
wxj
Browse files
Update convert.py 修复torch2.4.1 dtk25.4环境下转换失败的问题
parent
397d4ad5
Pipeline
#2195
passed with stage
Changes
1
Pipelines
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
5 additions
and
2 deletions
+5
-2
tools/checkpoint/convert.py
tools/checkpoint/convert.py
+5
-2
No files found.
tools/checkpoint/convert.py
View file @
5a1b96fa
...
...
@@ -137,10 +137,13 @@ def main():
args
=
parser
.
parse_args
()
queue
=
mp
.
Queue
(
maxsize
=
args
.
max_queue_size
)
ctx
=
mp
.
get_context
(
"spawn"
)
queue
=
ctx
.
Queue
(
maxsize
=
args
.
max_queue_size
)
# queue = mp.Queue(maxsize=args.max_queue_size)
print
(
"Starting saver..."
)
saver_proc
=
mp
.
Process
(
target
=
saver
.
save_checkpoint
,
args
=
(
queue
,
args
))
saver_proc
=
ctx
.
Process
(
target
=
saver
.
save_checkpoint
,
args
=
(
queue
,
args
))
# saver_proc = mp.Process(target=saver.save_checkpoint, args=(queue, args))
saver_proc
.
start
()
print
(
"Starting loader..."
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment