"git@developer.sourcefind.cn:gaoqiong/composable_kernel.git" did not exist on "5d90769e581f2b3bcbfaa9a1f09a5fb0c51dc1b2"
Unverified Commit f86c7005 authored by Scarlett Li's avatar Scarlett Li Committed by GitHub
Browse files

fix a small problem in example code (#2110)

* Update mnist.py for pytorch

it's weird putting pytorch's tmp data in a folder called /tmp/tensorflow...
parent 5f5f86c6
......@@ -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',
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment