1 type derived from TrainerBase
Microsoft.ML.TorchSharp (1)
NasBert\NasBertTrainer.cs (1)
176private protected abstract class NasBertTrainerBase : TrainerBase
5 references to TrainerBase
Microsoft.ML.TorchSharp (5)
NasBert\NerTrainer.cs (1)
104private protected override TrainerBase CreateTrainer(TorchSharpBaseTrainer<VBuffer<uint>, TargetType> parent, IChannel ch, IDataView input)
NasBert\SentenceSimilarityTrainer.cs (1)
102private protected override TrainerBase CreateTrainer(TorchSharpBaseTrainer<float, float> parent, IChannel ch, IDataView input)
NasBert\TextClassificationTrainer.cs (1)
102private protected override TrainerBase CreateTrainer(TorchSharpBaseTrainer<uint, long> parent, IChannel ch, IDataView input)
TorchSharpBaseTrainer.cs (2)
107var trainer = CreateTrainer(this, ch, input); 130private protected abstract TrainerBase CreateTrainer(TorchSharpBaseTrainer<TLabelCol, TTargetsCol> parent, IChannel ch, IDataView input);