Struct tensorflow_proto::tensorflow::tpu::OnlineYogiParameters

pub struct OnlineYogiParameters {
    pub l1: f32,
    pub l2: f32,
    pub beta2: f32,
    pub initial_v: f32,
    pub initial_linear: f32,
    pub activation: Option<Activation>,
}

The online Yogi optimizer does not implement hyper-parameter update; use the dynamic learning rate feature instead, setting the learning rate to: user learning_rate * sqrt(1 - beta2^t) / (1 - beta1^t) Here, t is the current timestep.

https://papers.nips.cc/paper/8186-adaptive-methods-for-nonconvex-optimization.pdf plus some extensions based on FTRL.

Note that the code by default implements the lazy version of online Yogi.

Fields

l1: f32

The L1 regularization parameter (used analogously to the one in FTRL).

l2: f32

The L2 regularization parameter (used analogously to the one in FTRL).

beta2: f32

\beta_2 from Algorithm 2 in the paper.

initial_v: f32

Initial value of V variable in paper.

initial_linear: f32

Initial value of linear variable in FTRL.

activation: Option<Activation>

Activation to use to replace sign function in v_t update in Algorithm 2 of paper.

Trait Implementations

impl Clone for OnlineYogiParameters

pub fn clone(&self) -> OnlineYogiParameters

Returns a copy of the value.

pub fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for OnlineYogiParameters

pub fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for OnlineYogiParameters

pub fn default() -> Self

Returns the "default value" for a type.

impl Message for OnlineYogiParameters

pub fn encode_raw<B>(&self, buf: &mut B) where
    B: BufMut, 

pub fn merge_field<B>(
    &mut self,
    tag: u32,
    wire_type: WireType,
    buf: &mut B,
    ctx: DecodeContext
) -> Result<(), DecodeError> where
    B: Buf, 

pub fn encoded_len(&self) -> usize

Returns the encoded length of the message without a length delimiter.

pub fn clear(&mut self)

Clears the message, resetting all fields to their default.

pub fn encode<B>(&self, buf: &mut B) -> Result<(), EncodeError> where
    B: BufMut, 

Encodes the message to a buffer.

pub fn encode_length_delimited<B>(&self, buf: &mut B) -> Result<(), EncodeError> where
    B: BufMut, 

Encodes the message with a length-delimiter to a buffer.

pub fn decode<B>(buf: B) -> Result<Self, DecodeError> where
    Self: Default,
    B: Buf, 

Decodes an instance of the message from a buffer.

pub fn decode_length_delimited<B>(buf: B) -> Result<Self, DecodeError> where
    Self: Default,
    B: Buf, 

Decodes a length-delimited instance of the message from the buffer.

pub fn merge<B>(&mut self, buf: B) -> Result<(), DecodeError> where
    B: Buf, 

Decodes an instance of the message from a buffer, and merges it into self.

pub fn merge_length_delimited<B>(&mut self, buf: B) -> Result<(), DecodeError> where
    B: Buf, 

Decodes a length-delimited instance of the message from buffer, and merges it into self.

impl PartialEq<OnlineYogiParameters> for OnlineYogiParameters

pub fn eq(&self, other: &OnlineYogiParameters) -> bool

This method tests for self and other values to be equal, and is used by ==.

pub fn ne(&self, other: &OnlineYogiParameters) -> bool

This method tests for !=.

impl StructuralPartialEq for OnlineYogiParameters