cossio / RestrictedBoltzmannMachines.jl

Train and sample Restricted Boltzmann machines in Julia
MIT License
14 stars 3 forks source link

dReLU energy at zero #4

Closed cossio closed 2 years ago

cossio commented 2 years ago

If the unit has zero activation, the dReLU energy should give zero.

Currently, since the dReLU energy is implemented by calling the ReLU energy, it gives Inf.

This should be corrected.

Edit: Actually this works fine, because the ReLU energy at zero is zero. The ReLU energy gives Inf only if x < 0.