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