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)
同时进行