importtorch# 设置随机数种子seed=42torch.manual_seed(seed) 1. 2. 3. 4. 5. 2. 设置CUDA随机数种子 如果你使用的是GPU训练,还需要设置CUDA的随机数种子,可以通过torch.cuda.manual_seed(seed)来实现。 # 设置CUDA随机数种子torch.cuda.manual_seed(seed) 1. 2. 3. 设置Python的随机数种子 除了设置PyT...
Pytorch torch.manual_seed(seed)# 为CPU设置随机种子torch.cuda.manual_seed(seed)# 为当前GPU设置随机种子torch.cuda.manual_seed_all(seed)# 为所有GPU设置随机种子 AI代码助手复制代码 Python & Numpy 如果读取数据的过程采用了随机预处理(如RandomCrop、RandomHorizontalFlip等),那么对python、numpy的随机数生成器...
importtorchimporttorch.nnasnnimportmatplotlib.pyplotaspltfromtoolsimportset_seedfromtorch.utils.tensorboardimportSummaryWriter set_seed(1)# 设置随机种子n_hidden =200max_iter =2000disp_interval =200lr_init =0.01defgen_data(num_data=10, x_range=(-1,1)): w =1.5train_x = torch.linspace(*x_ran...