Update convert.py

tools/checkpoint/convert.py

@@ -150,11 +150,14 @@ def main():
     args = parser.parse_args()
 
     # Initialize queue
-    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)
 
     # Start saver process.
     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()
 
     # Run loader.