Optimizer

1 write to Optimizer

Microsoft.ML.TorchSharp (1)

NasBert\Optimizers\Adam.cs (1)

18Optimizer = torch.optim.Adam(Parameters, options.LearningRate[0],

7 references to Optimizer

Microsoft.ML.TorchSharp (7)

NasBert\NasBertTrainer.cs (2)

190base.Optimizer = Optimizer.Optimizer; 192Optimizer.Optimizer,

NasBert\Optimizers\BaseOptimizer.cs (3)

35public double LearningRate => Optimizer.ParamGroups.ToArray()[0].LearningRate; 53Optimizer.step(); 61public void zero_grad() => Optimizer.zero_grad();

Roberta\QATrainer.cs (2)

171trainer.Optimizer.Optimizer.Dispose(); 221Optimizer.Optimizer,