Skip to content

Commit

Permalink
ESN reset
Browse files Browse the repository at this point in the history
  • Loading branch information
nschaetti committed Jun 18, 2018
1 parent 6b8c81b commit 6f059bf
Showing 1 changed file with 0 additions and 3 deletions.
3 changes: 0 additions & 3 deletions echotorch/nn/StackedESN.py
Original file line number Diff line number Diff line change
Expand Up @@ -128,9 +128,6 @@ def __init__(self, input_dim, hidden_dim, output_dim, leaky_rate=1.0, spectral_r
layer_input_scaling, layer_w, layer_w_in, layer_w_bias, None, layer_sparsity, layer_input_set,
layer_w_sparsity, layer_nonlin_func
))

# Apply spectral radius for Deep ESN
self.esn_layers[-1].w = layer_spectral_radius / echotorch.utils.deep_spectral_radius(w, layer_leaky_rate)
# end for

# Output layer
Expand Down

0 comments on commit 6f059bf

Please sign in to comment.