Commit 31dc58e9 authored by demianzhang's avatar demianzhang Committed by QuanluZhang
Browse files

fix example error (#1088)

parent f8ba4017
...@@ -7,6 +7,7 @@ import logging ...@@ -7,6 +7,7 @@ import logging
import math import math
import tempfile import tempfile
import time import time
import argparse
import tensorflow as tf import tensorflow as tf
from tensorflow.examples.tutorials.mnist import input_data from tensorflow.examples.tutorials.mnist import input_data
...@@ -20,7 +21,7 @@ class MnistNetwork(object): ...@@ -20,7 +21,7 @@ class MnistNetwork(object):
def __init__(self, params, feature_size = 784): def __init__(self, params, feature_size = 784):
config = [] config = []
for i in range(10): for i in range(4):
config.append(params['layer'+str(i)]) config.append(params['layer'+str(i)])
self.config = config self.config = config
self.feature_size = feature_size self.feature_size = feature_size
......
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