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