struct

ConstructionKit::ModelConfig

Inherits JSON::Serializable < Struct < Value < Object

Extract the model configuration from a validated pipeline graph. data_file is passed separately (workspace-level setting, not in the graph).

Constructors

new(data_file : String, seq_len : Int32, stream_dim : Int32, expert_specs : Array(ConstructionKit::ExpertSpec), router_type : String, router_epsilon : Float64, has_counter : Bool, learning_rate : Float64)
Source
new(pull : JSON::PullParser)
Source
new(*, __pull_for_json_serializable pull : JSON::PullParser)
Source

Instance methods

data_file
Source
data_file=(data_file : String)
Source
expert_specs
Source
expert_specs=(expert_specs : Array(ExpertSpec))
Source
has_counter
Source
has_counter=(has_counter : Bool)
Source
learning_rate
Source
learning_rate=(learning_rate : Float64)
Source
router_epsilon
Source
router_epsilon=(router_epsilon : Float64)
Source
router_type
Source
router_type=(router_type : String)
Source
seq_len
Source
seq_len=(seq_len : Int32)
Source
stream_dim
Source
stream_dim=(stream_dim : Int32)
Source