可复现性#
为了限制非确定性行为的来源,请在你的训练函数的顶部添加 ray.train.torch.enable_reproducibility()。
def train_func():
+ train.torch.enable_reproducibility()
model = NeuralNetwork()
model = train.torch.prepare_model(model)
...
警告
ray.train.torch.enable_reproducibility() 无法保证完全可复现的结果。若想了解更多,请阅读 PyTorch 关于随机性的说明。