18 references to NasBertTrainer
Microsoft.ML.TorchSharp (18)
NasBert\Models\BaseModel.cs (2)
12protected readonly NasBertTrainer.NasBertOptions Options; 24protected BaseModel(NasBertTrainer.NasBertOptions options)
NasBert\Models\ModelPrediction.cs (1)
19public ModelForPrediction(NasBertTrainer.NasBertOptions options, int padIndex, int symbolsCount, int numClasses)
NasBert\Models\NasBertModel.cs (1)
19public NasBertModel(NasBertTrainer.NasBertOptions options, int padIndex, int symbolsCount)
NasBert\Models\NerModel.cs (1)
18public NerModel(NasBertTrainer.NasBertOptions options, int padIndex, int symbolsCount, int numLabels)
NasBert\NasBertTrainer.cs (4)
167internal readonly NasBertTrainer.NasBertOptions BertOptions; 171BertOptions = options as NasBertTrainer.NasBertOptions; 413internal readonly NasBertTrainer.NasBertOptions BertOptions; 419internal NasBertTransformer(IHostEnvironment env, NasBertTrainer.NasBertOptions options, NasBertModel model, DataViewSchema.DetachedColumn labelColumn)
NasBert\NerTrainer.cs (1)
21using static Microsoft.ML.TorchSharp.NasBert.NasBertTrainer;
NasBert\Optimizers\Adam.cs (1)
15public Adam(NasBertTrainer.NasBertOptions options, IEnumerable<Parameter> parameters)
NasBert\Optimizers\BaseOptimizer.cs (3)
26public static BaseOptimizer GetOptimizer(NasBertTrainer.NasBertOptions options, IEnumerable<Parameter> parameters) 31protected NasBertTrainer.NasBertOptions Options { get; set; } 37protected BaseOptimizer(string name, NasBertTrainer.NasBertOptions options, IEnumerable<Parameter> parameters)
NasBert\SentenceSimilarityTrainer.cs (1)
19using static Microsoft.ML.TorchSharp.NasBert.NasBertTrainer;
NasBert\TextClassificationTrainer.cs (2)
20using static Microsoft.ML.TorchSharp.NasBert.NasBertTrainer; 63public class TextClassificationOptions : NasBertTrainer.NasBertOptions
Roberta\QATrainer.cs (1)
20using static Microsoft.ML.TorchSharp.NasBert.NasBertTrainer;