AutoFormerV2 Construtor

Definição

Namespace:

Microsoft.ML.TorchSharp.AutoFormerV2

Assembly:

Microsoft.ML.TorchSharp.dll

Pacote:

Microsoft.ML.TorchSharp v0.21.1

Inicializa uma nova instância da 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)

Parâmetros

numClasses

Int32

O número de categorias de objeto.

embedChannels

List<Int32>

Os canais de inserção, que controlam a escala do modelo.

depths

List<Int32>

O número de blocos, que controlam a escala do modelo.

numHeads

List<Int32>

O número de cabeças, que controlam a escala do modelo.

device

TorchSharp.torch.Device

O dispositivo em que a inferência do modelo.