Unverified Commit 6276b437 authored by Sylvain Gugger's avatar Sylvain Gugger
Browse files

Fix repo consistency

parent 09110577
......@@ -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)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment