Compartir a través de


AutoFormerV2 Constructor

Definición

Inicializa una nueva instancia de la clase 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)

Parámetros

numClasses
Int32

Número de categorías de objetos.

embedChannels
List<Int32>

Los canales de inserción, que controlan la escala del modelo.

depths
List<Int32>

Número de bloques, que controlan la escala del modelo.

numHeads
List<Int32>

Número de cabezales, que controlan la escala del modelo.

device
TorchSharp.torch.Device

Dispositivo donde se inferencia el modelo.

Se aplica a