Skip to content

Commit d5367e1

Browse files
author
Philippe Rémy
authored
Update README.md
1 parent cb4e5fb commit d5367e1

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

README.md

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -95,13 +95,12 @@ model.fit(x, y) # Keras model.
9595

9696
### Arguments
9797

98-
`TCN(nb_filters=64, kernel_size=2, nb_stacks=1, dilations=[1, 2, 4, 8, 16, 32], activation='norm_relu', padding='causal', use_skip_connections=True, dropout_rate=0.0, return_sequences=True, name='tcn')`
98+
`TCN(nb_filters=64, kernel_size=2, nb_stacks=1, dilations=[1, 2, 4, 8, 16, 32], padding='causal', use_skip_connections=True, dropout_rate=0.0, return_sequences=True, name='tcn')`
9999

100100
- `nb_filters`: Integer. The number of filters to use in the convolutional layers.
101101
- `kernel_size`: Integer. The size of the kernel to use in each convolutional layer.
102102
- `dilations`: List. A dilation list. Example is: [1, 2, 4, 8, 16, 32, 64].
103103
- `nb_stacks`: Integer. The number of stacks of residual blocks to use.
104-
- `activation`: String. The activations to use in the residual blocks (norm_relu, wavenet, relu...).
105104
- `padding`: String. The padding to use in the convolutions. 'causal' for a causal network (as in the original implementation) and 'same' for a non-causal network.
106105
- `use_skip_connections`: Boolean. If we want to add skip connections from input to each residual block.
107106
- `return_sequences`: Boolean. Whether to return the last output in the output sequence, or the full sequence.

0 commit comments

Comments
 (0)