Поделиться через


AutoFormerV2 Конструктор

Определение

Инициализирует новый экземпляр класса 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)

Параметры

numClasses
Int32

Количество категорий объектов.

embedChannels
List<Int32>

Каналы внедрения, управляющие масштабом модели.

depths
List<Int32>

Количество блоков, управляющих масштабом модели.

numHeads
List<Int32>

Количество голов, управляющих масштабом модели.

device
TorchSharp.torch.Device

Устройство, на котором выводится модель.

Применяется к