1 write to Device
Microsoft.ML.TorchSharp (1)
AutoFormerV2\ObjectDetectionTrainer.cs (1)
524
Device
= TorchUtils.InitializeDevice(env);
4 references to Device
Microsoft.ML.TorchSharp (4)
AutoFormerV2\ObjectDetectionTrainer.cs (4)
535
if (
Device
.type == DeviceType.CUDA)
905
var midTensor0 = torch.tensor(image.GetBGRPixels, device: _parent.
Device
);
921
var transMidTensor = torch.zeros(1, 3, image.Height + padH, image.Width + padW, device: _parent.
Device
);
923
var imageTensor = ObjectDetectionTrainer.Trainer.Normalize(transMidTensor, _parent.
Device
);