Udostępnij przez


AutoFormerV2 Konstruktor

Definicja

Inicjuje nowe wystąpienie klasy 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)

Parametry

numClasses
Int32

Liczba kategorii obiektów.

embedChannels
List<Int32>

Kanały osadzania, które kontrolują skalę modelu.

depths
List<Int32>

Liczba bloków, które kontrolują skalę modelu.

numHeads
List<Int32>

Liczba głowic, które kontrolują skalę modelu.

device
TorchSharp.torch.Device

Urządzenie, na którym wnioskowanie modelu.

Dotyczy