1 type derived from TorchSharpBaseTrainer
Microsoft.ML.TorchSharp (1)
NasBert\NasBertTrainer.cs (1)
164public abstract class NasBertTrainer<TLabelCol, TTargetsCol> : TorchSharpBaseTrainer<TLabelCol, TTargetsCol>
11 references to TorchSharpBaseTrainer
Microsoft.ML.TorchSharp (11)
NasBert\NasBertTrainer.cs (1)
185public NasBertTrainerBase(TorchSharpBaseTrainer<TLabelCol, TTargetsCol> parent, IChannel ch, IDataView input, string modelUrl) : base(parent, ch, input, modelUrl)
NasBert\NerTrainer.cs (2)
104private protected override TrainerBase CreateTrainer(TorchSharpBaseTrainer<VBuffer<uint>, TargetType> parent, IChannel ch, IDataView input) 121public Trainer(TorchSharpBaseTrainer<VBuffer<uint>, TargetType> parent, IChannel ch, IDataView input) : base(parent, ch, input, ModelUrlString)
NasBert\SentenceSimilarityTrainer.cs (2)
102private protected override TrainerBase CreateTrainer(TorchSharpBaseTrainer<float, float> parent, IChannel ch, IDataView input) 116public Trainer(TorchSharpBaseTrainer<float, float> parent, IChannel ch, IDataView input) : base(parent, ch, input, ModelUrlString)
NasBert\TextClassificationTrainer.cs (2)
102private protected override TrainerBase CreateTrainer(TorchSharpBaseTrainer<uint, long> parent, IChannel ch, IDataView input) 116public Trainer(TorchSharpBaseTrainer<uint, long> parent, IChannel ch, IDataView input) : base(parent, ch, input, ModelUrlString)
TorchSharpBaseTrainer.cs (4)
129private protected abstract TorchSharpBaseTransformer<TLabelCol, TTargetsCol> CreateTransformer(IHost host, TorchSharpBaseTrainer<TLabelCol, TTargetsCol>.Options options, Module model, DataViewSchema.DetachedColumn labelColumn); 130private protected abstract TrainerBase CreateTrainer(TorchSharpBaseTrainer<TLabelCol, TTargetsCol> parent, IChannel ch, IDataView input); 138protected readonly TorchSharpBaseTrainer<TLabelCol, TTargetsCol> Parent; 145public TrainerBase(TorchSharpBaseTrainer<TLabelCol, TTargetsCol> parent, IChannel ch, IDataView input, string modelUrl)