Reproducibility#

To limit sources of nondeterministic behavior, add ray.train.torch.enable_reproducibility() to the top of your training function.

 def train_func():
+    train.torch.enable_reproducibility()

     model = NeuralNetwork()
     model = train.torch.prepare_model(model)

     ...

Warning

ray.train.torch.enable_reproducibility() can’t guarantee completely reproducible results across executions. To learn more, read the PyTorch notes on randomness.