Skip to content

Commit

Permalink
duplication of 'large number' in warning
Browse files Browse the repository at this point in the history
  • Loading branch information
turnmanh committed Feb 19, 2025
1 parent 18f92b1 commit 9d7f5b3
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions sbi/inference/posteriors/mcmc_posterior.py
Original file line number Diff line number Diff line change
Expand Up @@ -452,11 +452,11 @@ def sample_batched(
num_chains_extended = batch_size * num_chains
if num_chains_extended > 100:
warnings.warn(
"Note that for batched sampling, we use num_chains many chains for each"
" x in the batch. With the given settings, this results in a large "
f"number large number of chains ({num_chains_extended}), which can be "
"slow and memory-intensive for vectorized MCMC. Consider reducing the "
"number of chains.",
"Note that for batched sampling, we use num_chains many chains "
"for each x in the batch. With the given settings, this results "
f"in a large number of chains ({num_chains_extended}), which can "
"be slow and memory-intensive for vectorized MCMC. Consider "
"reducing the number of chains or batch size.",
stacklevel=2,
)
init_strategy_parameters["num_return_samples"] = num_chains_extended
Expand Down

0 comments on commit 9d7f5b3

Please sign in to comment.