alexnet dataset seed

pull/6384/head
wilfChen 4 years ago
parent db0c0caf10
commit b6b0f4c68d

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

Loading…
Cancel
Save