Skip to content

GitLab

Commit c94996c7 authored by Lee's avatar Lee Committed by chicm-ms
Browse files

Add different tuner config files for config_test (#760) 

Add integration test cases for tuners.
parent 5f8ffcd5
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tools/nni_annotation/testcase/annotated/mnist.py
View file @ c94996c7
...@@ -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 @ c94996c7
...@@ -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