Ian González comments

Repositories
Issues
Comments

Results 3 comments of


                                            Ian González

Dropout with rate: 1 raises on binary backend

Hey @seanmor5 , I've found where the division by zero is happening. It is [here](https://github.com/elixir-nx/axon/blob/1a5c0fb765c78b97db0eeda4fda3b7dc65b1f173/lib/axon/layers.ex#L1508) in the dropout definition: `out = Nx.select(mask, input / keep_prob, Nx.tensor(0, type: Nx.type(input)))` I'm a...

Dropout with rate: 1 raises on binary backend

Sure, @seanmor5 , thanks for the guidance! I'll make a PR.

Raise helpful custom error when a user attempts to call functions in constants

Hi! Could I give it a try given it's been a bit since there was some activity with the issue? :)