Skip to content
Snippets Groups Projects
Unverified Commit c3c2683a authored by Alexander Seiler's avatar Alexander Seiler Committed by GitHub
Browse files

Correct some typos (#62)

parent c0ffb6ed
No related branches found
No related tags found
No related merge requests found
...@@ -58,7 +58,7 @@ class DINOLoss(nn.Module): ...@@ -58,7 +58,7 @@ class DINOLoss(nn.Module):
Q /= torch.sum(Q, dim=0, keepdim=True) Q /= torch.sum(Q, dim=0, keepdim=True)
Q /= B Q /= B
Q *= B # the colomns must sum to 1 so that Q is an assignment Q *= B # the columns must sum to 1 so that Q is an assignment
return Q.t() return Q.t()
def forward(self, student_output_list, teacher_out_softmaxed_centered_list): def forward(self, student_output_list, teacher_out_softmaxed_centered_list):
......
...@@ -87,7 +87,7 @@ class iBOTPatchLoss(nn.Module): ...@@ -87,7 +87,7 @@ class iBOTPatchLoss(nn.Module):
Q /= torch.sum(Q, dim=0, keepdim=True) Q /= torch.sum(Q, dim=0, keepdim=True)
Q /= B Q /= B
Q *= B # the colomns must sum to 1 so that Q is an assignment Q *= B # the columns must sum to 1 so that Q is an assignment
return Q.t() return Q.t()
def forward(self, student_patch_tokens, teacher_patch_tokens, student_masks_flat): def forward(self, student_patch_tokens, teacher_patch_tokens, student_masks_flat):
......
...@@ -57,7 +57,7 @@ def get_sha(): ...@@ -57,7 +57,7 @@ def get_sha():
sha = _run(["git", "rev-parse", "HEAD"]) sha = _run(["git", "rev-parse", "HEAD"])
subprocess.check_output(["git", "diff"], cwd=cwd) subprocess.check_output(["git", "diff"], cwd=cwd)
diff = _run(["git", "diff-index", "HEAD"]) diff = _run(["git", "diff-index", "HEAD"])
diff = "has uncommited changes" if diff else "clean" diff = "has uncommitted changes" if diff else "clean"
branch = _run(["git", "rev-parse", "--abbrev-ref", "HEAD"]) branch = _run(["git", "rev-parse", "--abbrev-ref", "HEAD"])
except Exception: except Exception:
pass pass
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment