2 types derived from NasBertModel
Microsoft.ML.TorchSharp (2)
NasBert\Models\ModelPrediction.cs (1)
10internal sealed class ModelForPrediction : NasBertModel
NasBert\Models\NerModel.cs (1)
10internal class NerModel : NasBertModel
8 references to NasBertModel
Microsoft.ML.TorchSharp (8)
NasBert\NasBertTrainer.cs (4)
181public new NasBertModel Model; 206NasBertModel model; 419internal NasBertTransformer(IHostEnvironment env, NasBertTrainer.NasBertOptions options, NasBertModel model, DataViewSchema.DetachedColumn labelColumn) 608return (Parent.Model as NasBertModel).forward(inputTensor);
NasBert\NerTrainer.cs (2)
111return new NerTransformer(host, options as NasBertOptions, model as NasBertModel, labelColumn); 258internal NerTransformer(IHostEnvironment env, NasBertOptions options, NasBertModel model, DataViewSchema.DetachedColumn labelColumn) : base(env, options, model, labelColumn)
NasBert\TextClassificationTrainer.cs (2)
109return new TextClassificationTransformer(host, options as NasBertOptions, model as NasBertModel, labelColumn); 194internal TextClassificationTransformer(IHostEnvironment env, NasBertOptions options, NasBertModel model, DataViewSchema.DetachedColumn labelColumn) : base(env, options, model, labelColumn)