可重复性#
为了限制非确定性行为,请在训练函数的顶部添加 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 关于随机性的说明。