Skip to content

Commit dd56265

Browse files
dkazancmfep
authored andcommitted
update to defaults
1 parent e33286e commit dd56265

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

httomolibgpu/prep/stripe.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -621,10 +621,11 @@ def _repair_memory_fragmentation_if_needed(fragmentation_threshold: float = 0.2)
621621
if (total / pool.used_bytes()) - 1 > fragmentation_threshold:
622622
pool.free_all_blocks()
623623

624+
624625
def remove_stripe_fw(
625626
data: cp.ndarray,
626-
sigma: float = 1,
627-
wname: str = "sym16",
627+
sigma: float = 2,
628+
wname: str = "db5",
628629
level: Optional[int] = None,
629630
calc_peak_gpu_mem: bool = False,
630631
) -> cp.ndarray:
@@ -639,7 +640,7 @@ def remove_stripe_fw(
639640
sigma : float
640641
Damping parameter in Fourier space.
641642
wname : str
642-
Type of the wavelet filter: select from 'haar', 'db4', 'sym5', 'sym16' 'bior4.4'.
643+
Type of the wavelet filter: select from 'db5', 'db7', 'haar', 'sym5', 'sym16' 'bior4.4'.
643644
level : int, optional
644645
Number of discrete wavelet transform levels.
645646
calc_peak_gpu_mem: str:
@@ -693,7 +694,7 @@ def remove_stripe_fw(
693694
mem_stack.malloc(2 * np.prod(fcV_shape) * np.float32().itemsize)
694695
mem_stack.malloc(2 * fcV_bytes)
695696

696-
fft_dummy = cp.empty(fcV_shape, dtype='float32')
697+
fft_dummy = cp.empty(fcV_shape, dtype="float32")
697698
fft_plan = get_fft_plan(fft_dummy)
698699
fft_plan_size = fft_plan.work_area.mem.size
699700
del fft_dummy

0 commit comments

Comments
 (0)