Edit by browser
This commit is contained in:
parent
a6a633cf5e
commit
3aa87bf5e2
|
@ -180,7 +180,7 @@ if __name__ == "__main__":
|
|||
momentum = 0.00001
|
||||
epoch_size = 1
|
||||
########## end ##########
|
||||
mnist_path = "./MNIST_Data"
|
||||
mnist_path = "/opt/MNIST_Data"
|
||||
net_loss = SoftmaxCrossEntropyWithLogits(is_grad=False, sparse=True, reduction='mean')
|
||||
repeat_size = 1
|
||||
network = LeNet5()
|
||||
|
|
Loading…
Reference in New Issue