1 write to numClasses
Microsoft.ML.TorchSharp (1)
AutoFormerV2\RetinaHead.cs (1)
48
this.
numClasses
= numClasses;
1 reference to numClasses
Microsoft.ML.TorchSharp (1)
AutoFormerV2\RetinaHead.cs (1)
84
clsOutput = clsOutput.contiguous().view(clsOutput.shape[0], -1, this.
numClasses
);