2 writes to TopKAnswers
Microsoft.ML.TorchSharp (2)
Roberta\QATrainer.cs (2)
136TopKAnswers = topk, 694TopKAnswers = ctx.Reader.ReadInt32()
3 references to TopKAnswers
Microsoft.ML.TorchSharp (3)
Roberta\QATrainer.cs (3)
659ctx.Writer.Write(Options.TopKAnswers); 832TensorCacher outputCacher = new TensorCacher(_parent.Options.TopKAnswers); 920var topKSpans = MetricUtils.ComputeTopKSpansWithScore(logits, _parent.Options.TopKAnswers, questionLength, contextLength);