Update davit.py

pull/1630/head
Fredo Guan 2 years ago
parent 42f86c8c09
commit 45da04d620

@ -38,7 +38,7 @@ class MySequential(nn.Sequential):
def forward(self, inputs : Tensor, size : Tuple[int, int]):
for module in self:
inputs, size = module(inputs, size)
(inputs, size) = module(inputs, size)
return inputs
class ConvPosEnc(nn.Module):

Loading…
Cancel
Save