Skip to content

GitLab

"git@developer.sourcefind.cn:gaoqiong/flash-attention.git" did not exist on "8f4cd4c16bc3143b6a2aa3cecbcc8dc8d89dff9e"
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
Hide 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