Skip to content

GitLab

"...api/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "8b99f7e1571144afe7ed45619b04a482076f7e18"
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
Hide 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
...@@ -137,6 +138,8 @@ def _reset(): ...@@ -137,6 +138,8 @@ def _reset():
nni.runtime.platform.test._last_metric = None nni.runtime.platform.test._last_metric = None
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():
......
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