Share via

AutoFormerV2 Constructor

  • Reference

Definition

Namespace:
Microsoft.ML.TorchSharp.AutoFormerV2
Assembly:
Microsoft.ML.TorchSharp.dll
Package:
Microsoft.ML.TorchSharp v0.21.1

Initializes a new instance of the AutoFormerV2 class.

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)

Parameters

numClasses
Int32

The number of object categories.

embedChannels
List<Int32>

The embedding channels, which control the scale of model.

depths
List<Int32>

The number of blocks, which control the scale of model.

numHeads
List<Int32>

The number of heads, which control the scale of model.

device
TorchSharp.torch.Device

The device where the model inference.

Applies to