1 write to Device
Microsoft.ML.TorchSharp (1)
Roberta\QATrainer.cs (1)
575
Device
= TorchUtils.InitializeDevice(env);
2 references to Device
Microsoft.ML.TorchSharp (2)
Roberta\QATrainer.cs (2)
582
if (
Device
.type == DeviceType.CUDA)
861
var srcTensor = torch.tensor((new[] { 0 /* InitToken */ }).Concat(questionTokenId).Concat(new[] { 2 /* SeparatorToken */ }).Concat(contextTokenId).ToList(), device: _parent.
Device
);