pub enum InferenceActivation {
ReLU,
Tanh,
}Expand description
Activation function type for inference
Selected at model export time and serialized into the .json model file.
Applied element-wise inside InferenceModel::forward on the hidden
layers, hence the deliberately small variant set.
Variants§
ReLU
Rectified linear unit, max(x, 0). Matches PPO’s default hidden
activation when training scripts export with --activation relu.
Tanh
Hyperbolic tangent, x.tanh(). Used as the implicit default when a
serialized model omits the activation field (see the serde
default attribute on InferenceModel::activation).
Trait Implementations§
Source§impl Clone for InferenceActivation
impl Clone for InferenceActivation
Source§fn clone(&self) -> InferenceActivation
fn clone(&self) -> InferenceActivation
Returns a duplicate of the value. Read more
1.0.0 (const: unstable) · Source§fn clone_from(&mut self, source: &Self)
fn clone_from(&mut self, source: &Self)
Performs copy-assignment from
source. Read moreimpl Copy for InferenceActivation
Source§impl Debug for InferenceActivation
impl Debug for InferenceActivation
Source§impl<'de> Deserialize<'de> for InferenceActivation
impl<'de> Deserialize<'de> for InferenceActivation
Source§fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>where
__D: Deserializer<'de>,
fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>where
__D: Deserializer<'de>,
Deserialize this value from the given Serde deserializer. Read more
Auto Trait Implementations§
impl Freeze for InferenceActivation
impl RefUnwindSafe for InferenceActivation
impl Send for InferenceActivation
impl Sync for InferenceActivation
impl Unpin for InferenceActivation
impl UnsafeUnpin for InferenceActivation
impl UnwindSafe for InferenceActivation
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
Source§impl<T> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
impl<T> DeserializeOwned for Twhere
T: for<'de> Deserialize<'de>,
Source§impl<T> Instrument for T
impl<T> Instrument for T
Source§fn instrument(self, span: Span) -> Instrumented<Self>
fn instrument(self, span: Span) -> Instrumented<Self>
Source§fn in_current_span(self) -> Instrumented<Self>
fn in_current_span(self) -> Instrumented<Self>
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read more