AutoFormerV2 Constructeur

Définition

Espace de noms:

Microsoft.ML.TorchSharp.AutoFormerV2

Assembly:

Microsoft.ML.TorchSharp.dll

Paquet:

Microsoft.ML.TorchSharp v0.21.1

Initialise une nouvelle instance de la classe AutoFormerV2.

public AutoFormerV2 (int numClasses, System.Collections.Generic.List<int> embedChannels, System.Collections.Generic.List<int> depths, System.Collections.Generic.List<int> numHeads, TorchSharp.torch.Device device = default);

new Microsoft.ML.TorchSharp.AutoFormerV2.AutoFormerV2 : int * System.Collections.Generic.List<int> * System.Collections.Generic.List<int> * System.Collections.Generic.List<int> * TorchSharp.torch.Device -> Microsoft.ML.TorchSharp.AutoFormerV2.AutoFormerV2

Public Sub New (numClasses As Integer, embedChannels As List(Of Integer), depths As List(Of Integer), numHeads As List(Of Integer), Optional device As torch.Device = Nothing)

Paramètres

numClasses

Int32

Nombre de catégories d’objets.

embedChannels

List<Int32>

Canaux d’incorporation, qui contrôlent l’échelle du modèle.

depths

List<Int32>

Nombre de blocs qui contrôlent l’échelle du modèle.

numHeads

List<Int32>

Nombre de têtes qui contrôlent l’échelle du modèle.

device

TorchSharp.torch.Device

Appareil où l’inférence de modèle.