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