1 write to PredictionHead
Microsoft.ML.TorchSharp (1)
NasBert\Models\ModelPrediction.cs (1)
22
PredictionHead
= new PredictionHead(
3 references to PredictionHead
Microsoft.ML.TorchSharp (3)
NasBert\Models\ModelPrediction.cs (3)
15
public override BaseHead GetHead() =>
PredictionHead
;
35
x =
PredictionHead
.call(x);
45
PredictionHead
.Dispose();