Edit by browser
This commit is contained in:
parent
1e8046f231
commit
a6a633cf5e
|
@ -167,7 +167,7 @@ if __name__ == "__main__":
|
|||
lr = 0.001
|
||||
momentum = 0.001
|
||||
epoch_size = 1
|
||||
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