复现代码需要注意的设置 

if all_args.cuda_deterministic:
    torch.backends.cudnn.benchmark = False
    torch.backends.cudnn.deterministic = True

注意需要设置 cuda_deterministic == True
           cudnn.benchmark == False


posted @ 2021-11-26 11:50  呦呦南山  阅读(80)  评论(0)  编辑  收藏  举报