zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-08-22 04:56:41 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

reformatted by pre-commit black

Browse Source
This commit is contained in:
vidit 2023-12-24 04:53:02 +05:30
parent 4d3c451b3f
commit 6c3c16e07b
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
python/mlx/nn/layers/dropout.py
View File

@ -88,5 +88,3 @@ class Dropout2d(Module):
mask = mx.random.bernoulli(p=self._p_1, shape=mask_shape) mask = mx.random.bernoulli(p=self._p_1, shape=mask_shape)
return (1 / self._p_1) * mask * x return (1 / self._p_1) * mask * x