diff --git a/tools/checkpoint/convert.py b/tools/checkpoint/convert.py index 935613b143..51e453bbbe 100644 --- a/tools/checkpoint/convert.py +++ b/tools/checkpoint/convert.py @@ -136,7 +136,7 @@ def main(): saver.add_arguments(parser) args = parser.parse_args() - + mp.set_start_method('spawn') queue = mp.Queue(maxsize=args.max_queue_size) print("Starting saver...")