def __init__(self):
super(Fuse, self).__init__()
self.conv = nn.Conv3d(3,
2, (3, 5, 2),
stride=(2, 1, 1),
padding=(3, 2, 0),
bias=False)
self.bn = nn.BatchNorm3d(2)
def __init__(self):
super(TraceModel, self).__init__()
self.conv1 = nn.Conv1d(16, 33, 3, stride=2)
self.conv2 = nn.Conv2d(16,
33, (3, 5),
stride=(2, 1),
padding=(4, 2),
dilation=(3, 1))
self.conv3 = nn.Conv3d(16,
33, (3, 5, 2),
stride=(2, 1, 1),
padding=(4, 2, 0))