Skip to content

Commit 3240045

Browse files
committed
rm redundant code
Signed-off-by: Can-Zhao <canz@nvidia.com>
1 parent eb631d1 commit 3240045

File tree

1 file changed

+2
-6
lines changed

1 file changed

+2
-6
lines changed

generation/maisi/scripts/diff_model_infer.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -180,15 +180,11 @@ def run_inference(
180180

181181

182182
inferer = SlidingWindowInferer(
183-
roi_size=(
184-
min(output_size[0] // divisor // 4 * 3, 96),
185-
min(output_size[1] // divisor // 4 * 3, 96),
186-
min(output_size[2] // divisor // 4 * 3, 96),
187-
),
183+
roi_size=[80,80,80],
188184
sw_batch_size=1,
189185
progress=True,
190186
mode="gaussian",
191-
overlap=2.0 / 3.0,
187+
overlap=0.4,
192188
sw_device=device,
193189
device=device,
194190
)

0 commit comments

Comments
 (0)