flax
24bc7bd2 - Handle edge-case of rate==1.0 in Dropout layer.

Commit

3 years ago

Handle edge-case of rate==1.0 in Dropout layer. Though forward eval works, gradient of a dropout layer with rate==1.0 had NaNs due to select on NaN rescaling factor.

References

#2055 - Handle edge-case of rate==1.0 in Dropout layer.

Author

levskaya

Committer

levskaya

Parents

c54e1131

flax 24bc7bd2 - Handle edge-case of rate==1.0 in Dropout layer.

flax
24bc7bd2 - Handle edge-case of rate==1.0 in Dropout layer.