Skip to Content

rmsprop

View the code on GitHub

Structs

Struct: RMSprop

Fields

  • params: List[Array]

  • lr: SIMD[float32, 1]

  • alpha: SIMD[float32, 1]

  • eps: SIMD[float32, 1]

  • cache: List[Array]

Methods

__init__(out self, params: List[Array], lr: SIMD[float32, 1] = SIMD(#kgen.float_literal<1|100>), alpha: SIMD[float32, 1] = SIMD(#kgen.float_literal<99|100>), eps: SIMD[float32, 1] = SIMD(#kgen.float_literal<1|100000000>))
more details
Args
  • self: Self

  • params: List[Array]

  • lr: SIMD[float32, 1] (default: SIMD(#kgen.float_literal<1|100>))

  • alpha: SIMD[float32, 1] (default: SIMD(#kgen.float_literal<99|100>))

  • eps: SIMD[float32, 1] (default: SIMD(#kgen.float_literal<1|100000000>))

step(mut self)
more details
Args
  • self: Self

Functions

Last updated on