"src/vscode:/vscode.git/clone" did not exist on "06db4729b00bd2d6f5582fe8528014ffe106953a"
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
import math
import tempfile
import time
import argparse
import tensorflow as tf
from tensorflow.examples.tutorials.mnist import input_data
......@@ -20,7 +21,7 @@ class MnistNetwork(object):
def __init__(self, params, feature_size = 784):
config = []
for i in range(10):
for i in range(4):
config.append(params['layer'+str(i)])
self.config = config
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