File size: 303 Bytes
9c6594c
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
#!/Users/james/src/MMaDA/venv/bin/python

from deepspeed.nvme import sweep_main, generate_main, parse_sweep_arguments

if __name__ == '__main__':
    args = parse_sweep_arguments()
    print(f"Running DeepNVMe performance tuning on {args.nvme_dir}")
    sweep_main(args)
    generate_main(args.log_dir)