Skip to content

Commit

Permalink
Update convolution.py
Browse files Browse the repository at this point in the history
  • Loading branch information
ASEM000 committed Aug 3, 2023
1 parent fc24b38 commit d774a5e
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion serket/nn/convolution.py
Original file line number Diff line number Diff line change
Expand Up @@ -282,7 +282,6 @@ class Conv1D(ConvND):
groups: number of groups to use for grouped convolution.
key: key to use for initializing the weights. defaults to ``jax.random.PRNGKey(0)``.
dtype: dtype of the weights. defaults to ``jax.numpy.float32``
dtype: dtype to use for the weights and bias. defaults to ``jnp.float32``.
Example:
>>> import jax.numpy as jnp
Expand Down

0 comments on commit d774a5e

Please sign in to comment.