版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/ccbrid/article/details/85108243
1. gpu:
torch.cuda.manual_seed(args.seed)
2. cpu:
torch.manual_seed(args.seed)
3. random:
random.seed(args.seed)
4. numpy:
numpy.random.seed(args.seed)
同时进行
1. gpu:
torch.cuda.manual_seed(args.seed)
2. cpu:
torch.manual_seed(args.seed)
3. random:
random.seed(args.seed)
4. numpy:
numpy.random.seed(args.seed)
同时进行