2 writes to NumberOfClasses
Microsoft.ML.TorchSharp (2)
AutoFormerV2\ObjectDetectionTrainer.cs (2)
255
Parent.Option.
NumberOfClasses
= uniqueLabels.Count;
681
NumberOfClasses
= ctx.Reader.ReadInt32(),
5 references to NumberOfClasses
Microsoft.ML.TorchSharp (5)
AutoFormerV2\ObjectDetectionTrainer.cs (5)
217
Parent.Option.
NumberOfClasses
,
620
ctx.Writer.Write(Options.
NumberOfClasses
);
688
var model = new AutoFormerV2(options.
NumberOfClasses
,
767
info[0] = new DataViewSchema.DetachedColumn(_parent.Options.PredictedLabelColumnName, new VectorDataViewType(new KeyDataViewType(typeof(uint), _parent.Options.
NumberOfClasses
)), labelBuilder.ToAnnotations());
881
TensorCacher outputCacher = new TensorCacher(_parent.Options.
NumberOfClasses
);