摘要：
if all_args.cuda_deterministic: torch.backends.cudnn.benchmark = False torch.backends.cudnn.deterministic = True注意需要设置 cuda_deterministic == True cudn 阅读全文
posted @ 2021-11-26 11:50
呦呦南山
阅读(93)
评论(0)
推荐(0)
摘要：
在服务器上跑PyTorch 程序的列位不要把cpu占满修改这个环境变量降低你们的cpu使用率（因为pytorch默认是能用多少线程就用多少的） torch.set_num_threads(all_args.n_training_threads) 阅读全文
posted @ 2021-11-26 11:49
呦呦南山
阅读(610)
评论(0)
推荐(0)
浙公网安备 33010602011771号