Skip to content

Commit f32881f

Browse files
committed
fixing API
1 parent dba1517 commit f32881f

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

src/network3.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -74,10 +74,10 @@ def create_net(mini_batch_size=10, activation_fn=tanh):
7474
#ConvPoolLayer(image_shape=(mini_batch_size, 20, 12, 12), filter_shape=(40, 20, 5, 5), poolsize=(2, 2), activation_fn=activation_fn),
7575
#FullyConnectedLayer(n_in=40*4*4, n_out=100, mini_batch_size=mini_batch_size, activation_fn=activation_fn),
7676
#FullyConnectedLayer(n_in=784, n_out=100, mini_batch_size=mini_batch_size, activation_fn=activation_fn),
77-
#FullyConnectedLayer(n_in=20*12*12, n_out=100, mini_batch_size=mini_batch_size),
77+
FullyConnectedLayer(n_in=20*12*12, n_out=100),
7878
#FullyConnectedLayer(n_in=100, n_out=100, mini_batch_size=mini_batch_size, activation_fn=activation_fn),
79-
#SoftmaxLayer(n_in=100, n_out=10, mini_batch_size=mini_batch_size)], mini_batch_size)
80-
SoftmaxLayer(n_in=20*12*12, n_out=10)], mini_batch_size)
79+
SoftmaxLayer(n_in=100, n_out=10)], mini_batch_size)
80+
#SoftmaxLayer(n_in=20*12*12, n_out=10)], mini_batch_size)
8181

8282
#### Load the MNIST data
8383
def load_data_shared(filename="../data/mnist.pkl.gz"):
@@ -252,7 +252,7 @@ def set_inpt(self, inpt, mini_batch_size):
252252

253253
class FullyConnectedLayer():
254254

255-
def __init__(self, n_in, n_out, mini_batch_size=10, activation_fn=sigmoid):
255+
def __init__(self, n_in, n_out, activation_fn=sigmoid):
256256
self.n_in = n_in
257257
self.n_out = n_out
258258
self.activation_fn = activation_fn

0 commit comments

Comments
 (0)