class Object
Public Instance Methods
_get_activation_fn(activation)
click to toggle source
# File lib/secryst/transformer.rb, line 374 def _get_activation_fn(activation) if activation == "relu" return Torch::NN::F.method(:relu) elsif activation == "gelu" return Torch::NN::F.method(:gelu) end raise RuntimeError, "activation should be relu/gelu, not %s" % activation end