Skip to content

GitLab

"host/online_compile/hip_utility/hipoc_program.cpp" did not exist on "1264925422920f24b3bb4fa34f178e31a23c97b5"
Unverified Commit f908be2c authored by Zhenhua Han's avatar Zhenhua Han Committed by GitHub
Browse files

[retiarii] use fixed seeds for unit test of CGO engine (#4361)

parent 1abaf627
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
test/ut/retiarii/test_cgo_engine.py
View file @ f908be2c
...@@ -5,6 +5,7 @@ import unittest ...@@ -5,6 +5,7 @@ import unittest
import time import time
import torch import torch
import torch.nn as nn import torch.nn as nn
from pytorch_lightning.utilities.seed import seed_everything
from pathlib import Path from pathlib import Path
...@@ -138,6 +139,8 @@ def _reset(): ...@@ -138,6 +139,8 @@ def _reset():
nni.retiarii.integration_api._advisor = None nni.retiarii.integration_api._advisor = None
nni.retiarii.execution.api._execution_engine = None nni.retiarii.execution.api._execution_engine = None
seed_everything(42)
def _new_trainer(): def _new_trainer():
transform = transforms.Compose([transforms.ToTensor(), transforms.Normalize((0.1307,), (0.3081,))]) transform = transforms.Compose([transforms.ToTensor(), transforms.Normalize((0.1307,), (0.3081,))])
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment