From bf36006ff08e579bb2679f5b35acb5552afab94a Mon Sep 17 00:00:00 2001 From: Francis Lata Date: Mon, 20 Jan 2025 22:54:54 -0800 Subject: [PATCH] set seed --- examples/mlperf/model_train.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/examples/mlperf/model_train.py b/examples/mlperf/model_train.py index 5431d18842..d7d9c2feb7 100644 --- a/examples/mlperf/model_train.py +++ b/examples/mlperf/model_train.py @@ -407,6 +407,10 @@ def train_retinanet(): config["bs"] = bs = getenv("BS", 128) config["num_epochs"] = num_epochs = getenv("EPOCHS", 4) + if seed: + Tensor.manual_seed(seed) + np.random.seed(seed=seed) + # ** initialize wandb ** if (WANDB := getenv("WANDB")): import wandb