Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!34683 Add epoch number for lenet train 

Merge pull request !34683 from ZPaC/fix-full-ps-lenet-acc
This commit is contained in:
i-robot 2022-05-20 06:08:49 +00:00 committed by Gitee
parent c5d08781bb 4c2affdaad
commit 57216a7b80
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/st/frontend_compile_cache/run_lenet_ps.py
View File

@ -127,7 +127,7 @@ if __name__ == "__main__":
model = Model(network, net_loss, net_opt, metrics={"Accuracy": Accuracy()})
ds_train = create_dataset(os.path.join(DATASET_PATH, "train"), 32, 1)
model.train(1, ds_train, callbacks=[LossMonitor()], dataset_sink_mode=False)
model.train(3, ds_train, callbacks=[LossMonitor()], dataset_sink_mode=False)
ds_eval = create_dataset(os.path.join(DATASET_PATH, "test"), 32, 1)
acc = model.eval(ds_eval, dataset_sink_mode=False)

2
tests/st/ps/full_ps/test_full_ps_lenet.py
View File

@ -129,7 +129,7 @@ if __name__ == "__main__":
model = Model(network, net_loss, net_opt, metrics={"Accuracy": Accuracy()})
ds_train = create_dataset(os.path.join(dataset_path, "train"), 32, 1)
model.train(1, ds_train, callbacks=[LossMonitor()], dataset_sink_mode=False)
model.train(3, ds_train, callbacks=[LossMonitor()], dataset_sink_mode=False)
ds_eval = create_dataset(os.path.join(dataset_path, "test"), 32, 1)
acc = model.eval(ds_eval, dataset_sink_mode=False)