torch.set_num_threads(all_args.n_training_threads) 可降低CPU使用率

在服务器上跑PyTorch 程序的列位不要把cpu占满
修改这个环境变量降低你们的cpu使用率
（因为pytorch默认是能用多少线程就用多少的）

 

torch.set_num_threads(all_args.n_training_threads)