Skip to content

Commit e3173f8

Browse files
authored
Error in comment of log_weights.py
There was an error in line 77 comment instead of you there should be user
1 parent 78c3235 commit e3173f8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

common/log_weights.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,7 @@ def log_ebc_norms(
7474
: min(sample_size, emb_weight_tensor.shape[0])
7575
]
7676
# WARNING: .cpu() transfer executes malloc that may be the cause of memory leaks
77-
# Change sample_size if the you observe frequent OOM errors or remove weight logging.
77+
# Change sample_size if the user observe frequent OOM errors or remove weight logging.
7878
norms = emb_weight_tensor[sample_mask].cpu().norm(dim=1).to(torch.float32)
7979
logging.info(f"Norm shape before reduction: {norms.shape}", rank=-1)
8080
norms = norms.mean().to(torch.device(f"cuda:{dist.get_rank()}"))

0 commit comments

Comments
 (0)