可复现性#
为了限制非确定性行为的来源,请在您的训练函数顶部添加 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 关于随机性的注意事项。