Fix repo consistency

This commit is contained in:
Sylvain Gugger 2022-12-05 09:02:56 -05:00
parent 0911057744
commit 6276b437a6
No known key found for this signature in database
GPG Key ID: 4198F83D867EFC2E

View File

@ -181,8 +181,8 @@ class TimeSformerDropPath(nn.Module):
super().__init__()
self.drop_prob = drop_prob
def forward(self, x: torch.Tensor) -> torch.Tensor:
return drop_path(x, self.drop_prob, self.training)
def forward(self, hidden_states: torch.Tensor) -> torch.Tensor:
return drop_path(hidden_states, self.drop_prob, self.training)
def extra_repr(self) -> str:
return "p={}".format(self.drop_prob)