1 write to Parent
Microsoft.ML.TorchSharp (1)
AutoFormerV2\ObjectDetectionTrainer.cs (1)
206
Parent
= parent;
18 references to Parent
Microsoft.ML.TorchSharp (18)
AutoFormerV2\ObjectDetectionTrainer.cs (18)
213
Device = TorchUtils.InitializeDevice(
Parent
.Host);
217
Parent
.Option.NumberOfClasses,
226
Device = TorchUtils.InitializeDevice(
Parent
.Host);
235
learningRate:
Parent
.Option.InitLearningRate,
236
weight_decay:
Parent
.Option.WeightDecay);
240
LearningRateScheduler = MultiStepLR(Optimizer,
Parent
.Option.Steps);
245
var labelCol = input.GetColumn<VBuffer<uint>>(
Parent
.Option.LabelColumnName);
255
Parent
.Option.NumberOfClasses = uniqueLabels.Count;
261
var destDir = Path.Combine(((IHostEnvironmentInternal)
Parent
.Host).TempFilePath, "mlnet");
269
using (var ch = (
Parent
.Host as IHostEnvironment).Start("Ensuring model file is present."))
271
var ensureModel = ResourceManagerUtils.Instance.EnsureResourceAsync(
Parent
.Host, ch, ModelUrl, destFileName, destDir, timeout);
288
DataViewRowCursor cursor = input.GetRowCursor(input.Schema[
Parent
.Option.LabelColumnName], input.Schema[
Parent
.Option.BoundingBoxColumnName], input.Schema[
Parent
.Option.ImageColumnName]);
290
var boundingBoxGetter = cursor.GetGetter<VBuffer<float>>(input.Schema[
Parent
.Option.BoundingBoxColumnName]);
291
var imageGetter = cursor.GetGetter<MLImage>(input.Schema[
Parent
.Option.ImageColumnName]);
292
var labelGetter = cursor.GetGetter<VBuffer<uint>>(input.Schema[
Parent
.Option.LabelColumnName]);
355
if (Updates %
Parent
.Option.LogEveryNStep == 0)