1 write to Model
Microsoft.ML.TorchSharp (1)
AutoFormerV2\ObjectDetectionTrainer.cs (1)
216
Model
= new AutoFormerV2(
8 references to Model
Microsoft.ML.TorchSharp (8)
AutoFormerV2\ObjectDetectionTrainer.cs (8)
183
transformer = new ObjectDetectionTransformer(Host, Option, trainer.
Model
, new DataViewSchema.DetachedColumn(labelCol.Value));
223
Model
.load(GetModelPath(), false);
230
Model
.cuda();
234
Model
.parameters(),
297
Model
.train();
298
Model
.FreezeBN();
347
var (classification, regression, anchors) =
Model
.forward(imageTensor);
350
torch.nn.utils.clip_grad_norm_(
Model
.parameters(), 0.1);