1 implementation of Model
Microsoft.ML.GenAI.Core (1)
Pipeline\CausalLMPipeline.cs (1)
118
public nn.Module<CausalLMModelInput, CausalLMModelOutput>
Model
{ get; }
2 references to Model
Microsoft.ML.GenAI.Core (2)
Trainer\CasualLMSupervisedFineTuningTrainer.cs (2)
37
var optimizer = new Adam(_pipeline.
Model
.parameters(), lr: trainingOption.LearningRate);
58
var output = _pipeline.
Model
.forward(new CausalLMModelInput(inputIds, attentionMask: attentionMask, labels: labels, useCache: false));