每个 GPU 进程内限制一下 intra-op 线程，避免 8 进程各自开满把 CPU 搞爆（你 CPU 很多也建议控一下）

if args.torch_threads > 0:
    torch.set_num_threads(int(args.torch_threads))
    torch.set_num_interop_threads(max(1, int(args.torch_threads // 2)))

如果是给 sonnet 4.5 要求做这个任务, 就不会出现这么口语的句子, 太好玩了来分享下 我还发现 GPT 家的习惯给代码上->注释类型, claude 的如果没要求, 改写的时候会不加, 新写的会加 仔细检查代码, 还发现模型擅自主张给结果进行了降低精确度, 从 float32 改为 float16, 有点讨厌, 太自作主张了, 精确度低了结果会差很多, 没发现可就浪费时间了