Merge pull request #2182 from fcossio/patch-1

Credit earlier work with the same idea
2024-05-16 15:10:46 -07:00 · 2024-05-16 15:10:46 -07:00 · 6b9e38b589
parent cb0e4391be 9b11801cb4
commit 6b9e38b589
1 changed files with 1 additions and 1 deletions
--- a/timm/layers/patch_dropout.py
+++ b/timm/layers/patch_dropout.py
@ -6,7 +6,7 @@ import torch.nn as nn

 class PatchDropout(nn.Module):
    """
-    https://arxiv.org/abs/2212.00794
+    https://arxiv.org/abs/2212.00794 and https://arxiv.org/pdf/2208.07220
    """
    return_indices: torch.jit.Final[bool]