9 references to Conv2dBN
Microsoft.ML.TorchSharp (9)
AutoFormerV2\AutoFormerV2Block.cs (1)
64
this.local_conv = new
Conv2dBN
(inChannels, inChannels, kernalSize: localConvSize, stride: 1, padding: padding, groups: inChannels);
AutoFormerV2\MBConv.cs (3)
37
this.conv1 = new
Conv2dBN
(inChannels, hiddenChans, kernalSize: 1);
39
this.conv2 = new
Conv2dBN
(hiddenChans, hiddenChans, kernalSize: 3, stride: 1, padding: 1, groups: hiddenChans);
41
this.conv3 = new
Conv2dBN
(hiddenChans, outChannels, kernalSize: 1);
AutoFormerV2\PatchEmbed.cs (2)
31
this.seq.Add(new
Conv2dBN
(inChannels, embedChannels / 2, 3, 2, 1));
33
this.seq.Add(new
Conv2dBN
(embedChannels / 2, embedChannels, 3, 2, 1));
AutoFormerV2\PatchMerging.cs (3)
34
this.conv1 = new
Conv2dBN
(inChannels, outChannels, 1, 1, 0);
35
this.conv2 = new
Conv2dBN
(outChannels, outChannels, 3, 2, 1, groups: outChannels);
36
this.conv3 = new
Conv2dBN
(outChannels, outChannels, 1, 1, 0);