Skip to content

Commit 12a1fc8

Browse files
author
Philippe Remy
committed
remove batch norm
1 parent 3c39fe0 commit 12a1fc8

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tasks/mnist_pixel/main.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -9,10 +9,10 @@ def run_task():
99
model = compiled_tcn(return_sequences=False,
1010
num_feat=1,
1111
num_classes=10,
12-
nb_filters=25,
13-
kernel_size=7,
12+
nb_filters=20,
13+
kernel_size=6,
1414
dilations=[2 ** i for i in range(9)],
15-
nb_stacks=2,
15+
nb_stacks=1,
1616
max_len=x_train[0:1].shape[1],
1717
use_skip_connections=True)
1818

0 commit comments

Comments
 (0)