1 write to Option
Microsoft.ML.TorchSharp (1)
AutoFormerV2\ObjectDetectionTrainer.cs (1)
131Option = options;
32 references to Option
Microsoft.ML.TorchSharp (32)
AutoFormerV2\ObjectDetectionTrainer.cs (32)
174for (int i = 0; i < Option.MaxEpoch; i++) 181var labelCol = input.Schema.GetColumnOrNull(Option.LabelColumnName); 183transformer = new ObjectDetectionTransformer(Host, Option, trainer.Model, new DataViewSchema.DetachedColumn(labelCol.Value)); 217Parent.Option.NumberOfClasses, 235learningRate: Parent.Option.InitLearningRate, 236weight_decay: Parent.Option.WeightDecay); 240LearningRateScheduler = MultiStepLR(Optimizer, Parent.Option.Steps); 245var labelCol = input.GetColumn<VBuffer<uint>>(Parent.Option.LabelColumnName); 255Parent.Option.NumberOfClasses = uniqueLabels.Count; 288DataViewRowCursor cursor = input.GetRowCursor(input.Schema[Parent.Option.LabelColumnName], input.Schema[Parent.Option.BoundingBoxColumnName], input.Schema[Parent.Option.ImageColumnName]); 290var boundingBoxGetter = cursor.GetGetter<VBuffer<float>>(input.Schema[Parent.Option.BoundingBoxColumnName]); 291var imageGetter = cursor.GetGetter<MLImage>(input.Schema[Parent.Option.ImageColumnName]); 292var labelGetter = cursor.GetGetter<VBuffer<uint>>(input.Schema[Parent.Option.LabelColumnName]); 355if (Updates % Parent.Option.LogEveryNStep == 0) 462inputSchema.TryFindColumn(Option.LabelColumnName, out var labelCol); 464outColumns[Option.PredictedLabelColumnName] = new SchemaShape.Column(Option.PredictedLabelColumnName, SchemaShape.Column.VectorKind.VariableVector, 467outColumns[Option.PredictedBoundingBoxColumnName] = new SchemaShape.Column(Option.PredictedBoundingBoxColumnName, SchemaShape.Column.VectorKind.VariableVector, 470outColumns[Option.ScoreColumnName] = new SchemaShape.Column(Option.ScoreColumnName, SchemaShape.Column.VectorKind.VariableVector, 480if (!inputSchema.TryFindColumn(Option.LabelColumnName, out var labelCol)) 481throw Host.ExceptSchemaMismatch(nameof(inputSchema), "label", Option.LabelColumnName); 483throw Host.ExceptSchemaMismatch(nameof(inputSchema), "label", Option.LabelColumnName, 486if (!inputSchema.TryFindColumn(Option.BoundingBoxColumnName, out var boundingBoxCol)) 487throw Host.ExceptSchemaMismatch(nameof(inputSchema), "BoundingBox", Option.BoundingBoxColumnName); 489throw Host.ExceptSchemaMismatch(nameof(inputSchema), "BoundingBox", Option.BoundingBoxColumnName, 492if (!inputSchema.TryFindColumn(Option.ImageColumnName, out var imageCol)) 493throw Host.ExceptSchemaMismatch(nameof(inputSchema), "Image", Option.ImageColumnName); 495throw Host.ExceptSchemaMismatch(nameof(inputSchema), "Image", Option.ImageColumnName,