Skip to content

Commit c4fe072

Browse files
author
Alexander Ororbia
committed
minor mod to model-utils
1 parent 1c5164e commit c4fe072

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ngclearn/utils/model_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -657,7 +657,7 @@ def layer_normalize(x, shift=0., scale=1.):
657657
@jit
658658
def drop_out(dkey, data, rate=0.0):
659659
"""
660-
Applies a drop-out transform to an input matrix.
660+
Applies a drop-out transform (i.e., a random number of elements will be dropped to zero) to an input matrix.
661661
662662
Args:
663663
dkey: Jax randomness key for this operator

0 commit comments

Comments
 (0)