Skip to content

Commit

Permalink
remove display
Browse files Browse the repository at this point in the history
  • Loading branch information
cmarshak committed Jan 23, 2025
1 parent d872f1d commit 6739401
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions src/distmetrics/transformer.py
Original file line number Diff line number Diff line change
Expand Up @@ -299,7 +299,6 @@ def _estimate_logit_params_via_streamed_patches(
mininterval=2,
disable=(not tqdm_enabled),
dynamic_ncols=True,
display=True,
):
chip_mean, chip_logvar = model(patch_batch)
for k, (sy, sx) in enumerate(slices):
Expand Down Expand Up @@ -407,7 +406,6 @@ def _estimate_logit_params_via_folding(
mininterval=2,
disable=(not tqdm_enabled),
dynamic_ncols=True,
display=True,
):
# change last dimension from P**2 to P, P; use -1 because won't always have batch_size as 0th dimension
batch_s = slice(batch_size * i, batch_size * (i + 1))
Expand Down

0 comments on commit 6739401

Please sign in to comment.