Skip to content

GitLab

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

Fix repo consistency

parent 09110577
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
src/transformers/models/timesformer/modeling_timesformer.py
View file @ 6276b437
...@@ -181,8 +181,8 @@ class TimeSformerDropPath(nn.Module): ...@@ -181,8 +181,8 @@ class TimeSformerDropPath(nn.Module):
super().__init__() super().__init__()
self.drop_prob = drop_prob self.drop_prob = drop_prob
def forward(self, x: torch.Tensor) -> torch.Tensor: def forward(self, hidden_states: torch.Tensor) -> torch.Tensor:
return drop_path(x, self.drop_prob, self.training) return drop_path(hidden_states, self.drop_prob, self.training)
def extra_repr(self) -> str: def extra_repr(self) -> str:
return "p={}".format(self.drop_prob) 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