github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-29 03:00:14 -04:00
Code Issues Packages Projects Releases Wiki Activity

set seed

Browse Source
This commit is contained in:
Francis Lata
2025-01-20 22:54:54 -08:00
parent 5d9a604963
commit bf36006ff0
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/mlperf/model_train.py
View File

@@ -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