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)


Signed-off-by: default avatarAlexander Seiler <seileralex@gmail.com>
parent c0ffb6ed
No related branches found
No related tags found
No related merge requests found
......@@ -58,7 +58,7 @@ class DINOLoss(nn.Module):
Q /= torch.sum(Q, dim=0, keepdim=True)
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()
def forward(self, student_output_list, teacher_out_softmaxed_centered_list):
......
......@@ -87,7 +87,7 @@ class iBOTPatchLoss(nn.Module):
Q /= torch.sum(Q, dim=0, keepdim=True)
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()
def forward(self, student_patch_tokens, teacher_patch_tokens, student_masks_flat):
......
......@@ -57,7 +57,7 @@ def get_sha():
sha = _run(["git", "rev-parse", "HEAD"])
subprocess.check_output(["git", "diff"], cwd=cwd)
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"])
except Exception:
pass
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment