Skip to content

GitLab

Unverified Commit 1ee97350 authored by SparkSnail's avatar SparkSnail Committed by GitHub
Browse files

Merge pull request #129 from Microsoft/master 

Add different tuner config files for config_test (#760)
parents 754a354b c94996c7
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tools/nni_annotation/testcase/annotated/mnist.py
View file @ 1ee97350
...@@ -127,7 +127,7 @@ def bias_variable(shape): ...@@ -127,7 +127,7 @@ def bias_variable(shape):
def main(): def main():
data_dir = '/tmp/tensorflow/mnist/input_data' data_dir = '/tmp/tensorflow/mnist/input_data'
mnist = input_data.read_data_sets(data_dir, one_hot=True) mnist = input_data.read_data_sets(data_dir, one_hot=True)
logger.debug('Mnist download data down.') logger.debug('Mnist download data done.')
mnist_network = MnistNetwork() mnist_network = MnistNetwork()
mnist_network.build_network() mnist_network.build_network()
logger.debug('Mnist build network done.') logger.debug('Mnist build network done.')
......
tools/nni_annotation/testcase/usercode/mnist.py
View file @ 1ee97350
...@@ -137,7 +137,7 @@ def main(): ...@@ -137,7 +137,7 @@ def main():
# Import data # Import data
data_dir= '/tmp/tensorflow/mnist/input_data' data_dir= '/tmp/tensorflow/mnist/input_data'
mnist = input_data.read_data_sets(data_dir, one_hot=True) mnist = input_data.read_data_sets(data_dir, one_hot=True)
logger.debug('Mnist download data down.') logger.debug('Mnist download data done.')
# Create the model # Create the model
# Build the graph for the deep net # Build the graph for the deep net
......
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