jiangzhongxiang
/
MindSpore-competition
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Edit by browser

This commit is contained in:
任俊达 2020-10-23 14:24:30 +08:00
parent a6a633cf5e
commit 3aa87bf5e2
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
step2/step2.py
View File

@ -180,7 +180,7 @@ if __name__ == "__main__":
momentum = 0.00001 momentum = 0.00001
epoch_size = 1 epoch_size = 1
########## end ########## ########## end ##########
mnist_path = "./MNIST_Data" mnist_path = "/opt/MNIST_Data"
net_loss = SoftmaxCrossEntropyWithLogits(is_grad=False, sparse=True, reduction='mean') net_loss = SoftmaxCrossEntropyWithLogits(is_grad=False, sparse=True, reduction='mean')
repeat_size = 1 repeat_size = 1
network = LeNet5() network = LeNet5()