1 write to Model
Microsoft.ML.TorchSharp (1)
Roberta\QATrainer.cs (1)
204
Model
= new RobertaModelForQA(Parent.Option);
6 references to Model
Microsoft.ML.TorchSharp (6)
Roberta\QATrainer.cs (6)
173
transformer = new QATransformer(Host, Option, trainer.
Model
);
206
Model
.GetEncoder().load(GetModelPath());
213
Model
.cuda();
218
var parameters =
Model
.parameters().Where(p => p.requires_grad);
349
Model
.train();
354
var logits =
Model
.forward(srcTensor); //[batchsize, maxseqlen, 2]