diff --git a/step1/step1.py b/step1/step1.py index de5ca13..23bc1d9 100644 --- a/step1/step1.py +++ b/step1/step1.py @@ -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()