深度学习中随机种子的初始化

版权声明:本文为博主原创文章,未经博主允许不得转载。 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)

同时进行

猜你喜欢

转载自blog.csdn.net/ccbrid/article/details/85108243