Commit 5a1b96fa authored by wxj's avatar wxj
Browse files

Update convert.py 修复torch2.4.1 dtk25.4环境下转换失败的问题

parent 397d4ad5
Pipeline #2195 passed with stage
......@@ -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...")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment