我训练过Pytorch的模型。但是,当我推断相同的输入时,结果并不确定。
我在推断之前执行model.eval()并将随机种子设置如下。
def setup_seed(seed):
torch.manual_seed(seed)
torch.cuda.manual_seed(seed)
torch.cuda.manual_seed_all(seed)
np.random.seed(seed)
random.seed(seed)
torch.backends.cudnn.benchmark = False
torch.backends.cudnn.deterministic = True
有人有建议吗?谢谢。