(self)
| 105 | """ |
| 106 | |
| 107 | def __init__(self): |
| 108 | super(visualFrontend, self).__init__() |
| 109 | self.frontend3D = nn.Sequential( |
| 110 | nn.Conv3d(1, 64, kernel_size=(5,7,7), stride=(1,2,2), padding=(2,3,3), bias=False), |
| 111 | nn.BatchNorm3d(64, momentum=0.01, eps=0.001), |
| 112 | nn.ReLU(), |
| 113 | nn.MaxPool3d(kernel_size=(1,3,3), stride=(1,2,2), padding=(0,1,1)) |
| 114 | ) |
| 115 | self.resnet = ResNet() |
| 116 | return |
| 117 | |
| 118 | |
| 119 | def forward(self, inputBatch): |