7 references to PadIndex
Microsoft.ML.TorchSharp (7)
NasBert\NasBertTrainer.cs (3)
208
model = new NerModel(Parent.BertOptions, tokenizerModel.
PadIndex
, tokenizerModel.SymbolsCount, Parent.Option.NumberOfClasses);
210
model = new ModelForPrediction(Parent.BertOptions, tokenizerModel.
PadIndex
, tokenizerModel.SymbolsCount, Parent.Option.NumberOfClasses);
240
return DataUtils.CollateTokens(inputTensors, Tokenizer.RobertaModel().
PadIndex
, device: Device);
NasBert\NerTrainer.cs (1)
332
var model = new NerModel(options, tokenizerModel.
PadIndex
, tokenizerModel.SymbolsCount, options.NumberOfClasses);
NasBert\SentenceSimilarityTrainer.cs (1)
244
var model = new ModelForPrediction(options, tokenizerModel.
PadIndex
, tokenizerModel.SymbolsCount, options.NumberOfClasses);
NasBert\TextClassificationTrainer.cs (1)
268
var model = new ModelForPrediction(options, tokenizerModel.
PadIndex
, tokenizerModel.SymbolsCount, options.NumberOfClasses);
Roberta\QATrainer.cs (1)
352
srcTensor = PrepareBatchTensor(ref inputTensors, device: Device, Tokenizer.RobertaModel().
PadIndex
);