【发布时间】:2021-05-13 18:23:18
【问题描述】:
我的网络包含“torch.nn.MaxPool3d”,当根据 PyTorch 文档(版本 1.7 - https://pytorch.org/docs/stable/generated/torch.set_deterministic.html#torch.set_deterministic)打开 cudnn 确定性标志时,它会引发 RuntimeError,但是,当我插入代码“torch.backends.cudnn”时.deterministic=True' 在我的代码开头,没有 RuntimeError。为什么该代码不会引发 RuntimeError? 我想知道该代码是否能保证我的训练过程的确定性计算。
【问题讨论】:
标签: pytorch deterministic reproducible-research