3 overrides of CreateTargetsTensor
Microsoft.ML.TorchSharp (3)
NasBert\NerTrainer.cs (1)
136private protected override torch.Tensor CreateTargetsTensor(ref List<TargetType> targets, torch.Device device)
NasBert\SentenceSimilarityTrainer.cs (1)
128private protected override torch.Tensor CreateTargetsTensor(ref List<float> targets, torch.Device device)
NasBert\TextClassificationTrainer.cs (1)
129private protected override torch.Tensor CreateTargetsTensor(ref List<long> targets, torch.Device device)
2 references to CreateTargetsTensor
Microsoft.ML.TorchSharp (2)
TorchSharpBaseTrainer.cs (2)
259var targetsTensor = CreateTargetsTensor(ref targets, device: Device); 335var targetsTensor = CreateTargetsTensor(ref targets, device: Device);