package kaun

  1. Overview
  2. Docs
package kaun
Legend:
Page
Library
Module
Module type
Parameter
Class
Class type
Source

Module Optimizer.ScheduleSource

Sourcetype t = int -> float
Sourceval constant : float -> t
Sourceval exponential_decay : init_value:float -> decay_rate:float -> decay_steps:int -> t
Sourceval polynomial_decay : init_value:float -> end_value:float -> power:float -> decay_steps:int -> t
Sourceval cosine_decay : init_value:float -> decay_steps:int -> ?alpha:float -> unit -> t
Sourceval piecewise_constant : boundaries:(int * float) list -> t
Sourceval warmup_linear : init_value:float -> peak_value:float -> warmup_steps:int -> t
Sourceval warmup_cosine : init_value:float -> peak_value:float -> warmup_steps:int -> t
Sourceval join : t list -> boundaries:int list -> t