"...composable_kernel_rocm.git" did not exist on "a8fafc3fed19dcbb6f11d4b9dc32f61ad3e5c077"
Unverified Commit 4b4b8642 authored by Ashish Tawari's avatar Ashish Tawari Committed by GitHub
Browse files

Fix for stochastic depth decay rule in the TimeSformer implementation (#27875)

Update modeling_timesformer.py

Fixing typo to correct the stochastic depth decay rule
parent c0a354d8
......@@ -305,7 +305,7 @@ class TimesformerLayer(nn.Module):
] # stochastic depth decay rule
drop_path_rate = drop_path_rates[layer_index]
self.drop_path = TimeSformerDropPath(config.drop_path_rate) if drop_path_rate > 0.0 else nn.Identity()
self.drop_path = TimeSformerDropPath(drop_path_rate) if drop_path_rate > 0.0 else nn.Identity()
self.attention = TimeSformerAttention(config)
self.intermediate = TimesformerIntermediate(config)
self.output = TimesformerOutput(config)
......
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