| #!/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) | |