Skip to content

Commit

Permalink
fix a small problem in example code (microsoft#2110)
Browse files Browse the repository at this point in the history
* Update mnist.py for pytorch

it's weird putting pytorch's tmp data in a folder called /tmp/tensorflow...
  • Loading branch information
scarlett2018 authored Mar 2, 2020
1 parent 5f5f86c commit f86c700
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion examples/trials/mnist-pytorch/mnist.py
Original file line number Diff line number Diff line change
Expand Up @@ -128,7 +128,7 @@ def get_params():
# Training settings
parser = argparse.ArgumentParser(description='PyTorch MNIST Example')
parser.add_argument("--data_dir", type=str,
default='/tmp/tensorflow/mnist/input_data', help="data directory")
default='/tmp/pytorch/mnist/input_data', help="data directory")
parser.add_argument('--batch_size', type=int, default=64, metavar='N',
help='input batch size for training (default: 64)')
parser.add_argument("--hidden_size", type=int, default=512, metavar='N',
Expand Down

0 comments on commit f86c700

Please sign in to comment.