dgl里的gatconv是不是非确定性的呀,我固定了所有的随机种子,但输出结果依然在变
解决:
def set_seed(seed):random.seed(seed)os.environ['PYTHONHASHSEED'] = str(seed)np.random.seed(seed)torch.manual_seed(seed)torch.cuda.manual_seed(seed)torch.cuda.manual_seed_all(seed)dgl.random.seed(seed)dgl.seed(seed)torch.backends.cudnn.deterministic = Truetorch.backends.cudnn.benchmark = Falsetorch.backends.cudnn.enabled = Falseset_seed(SEED)
