1 write to _pipeline
Microsoft.ML.GenAI.Core (1)
Trainer\CasualLMSupervisedFineTuningTrainer.cs (1)
24_pipeline = pipeline;
3 references to _pipeline
Microsoft.ML.GenAI.Core (3)
Trainer\CasualLMSupervisedFineTuningTrainer.cs (3)
37var optimizer = new Adam(_pipeline.Model.parameters(), lr: trainingOption.LearningRate); 58var output = _pipeline.Model.forward(new CausalLMModelInput(inputIds, attentionMask: attentionMask, labels: labels, useCache: false)); 82yield return _pipeline;