1 reference to AutoFormerV2Block
Microsoft.ML.TorchSharp (1)
AutoFormerV2\BasicLayer.cs (1)
51
this.blocks.Add(new
AutoFormerV2Block
(inChannels: inChannels, numHeads: numHeads, windowSize: windowSize, shiftSize: (i % 2 == 0) ? 0 : (windowSize / 2), mlpRatio: mlpRatio, dropRatio: dropRatio, localConvSize: localConvSize, useShiftWindow: useShiftWindow, useInterpolate: useInterpolate));