From b6b0f4c68dc6fc12a086307f68e9bc88a5c2c7de Mon Sep 17 00:00:00 2001 From: wilfChen Date: Thu, 17 Sep 2020 09:32:29 +0800 Subject: [PATCH] alexnet dataset seed --- model_zoo/official/cv/alexnet/train.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/model_zoo/official/cv/alexnet/train.py b/model_zoo/official/cv/alexnet/train.py index 4f638aacb2..ce23307438 100644 --- a/model_zoo/official/cv/alexnet/train.py +++ b/model_zoo/official/cv/alexnet/train.py @@ -28,6 +28,7 @@ from src.alexnet import AlexNet from src.get_param_groups import get_param_groups import mindspore.nn as nn from mindspore.communication.management import init, get_rank +from mindspore import dataset as de from mindspore import context from mindspore import Tensor from mindspore.train import Model @@ -37,6 +38,7 @@ from mindspore.train.callback import ModelCheckpoint, CheckpointConfig, LossMoni from mindspore.common import set_seed set_seed(1) +de.config.set_seed(1) if __name__ == "__main__": parser = argparse.ArgumentParser(description='MindSpore AlexNet Example')