class

ConstructionKit::LossExec

Inherits ConstructionKit::ExecutableNode < Reference < Object

Constructors

new(id : String)
Source

Instance methods

backward(output_grads : Hash(String, MicroGPT::Mat)) : Hash(String, MicroGPT::Mat)

Loss node's backward returns the gradient for logits

Source
forward(inputs : Hash(String, Tensor)) : Hash(String, Tensor)

Forward pass: read named inputs, produce named outputs

Source
loss
Source
update(lr : Float64)

Update trainable parameters

Source