Skip to content

Commit

Permalink
Update models/vista3d/scripts/inferer.py
Browse files Browse the repository at this point in the history
Co-authored-by: Dr. Bruce Hashemian <3968947+drbeh@users.noreply.github.com>
Signed-off-by: Yiheng Wang <68361391+yiheng-wang-nv@users.noreply.github.com>
  • Loading branch information
yiheng-wang-nv and drbeh authored Nov 11, 2024
1 parent de8903a commit 721787f
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion models/vista3d/scripts/inferer.py
Original file line number Diff line number Diff line change
Expand Up @@ -101,7 +101,7 @@ def __call__(
)
else:
val_outputs = SlidingWindowInfererAdapt(
roi_size=self.roi_size, sw_batch_size=self.sw_batch_size, with_coord=True, padding_mode='replicate'
roi_size=self.roi_size, sw_batch_size=self.sw_batch_size, with_coord=True, padding_mode="replicate"
)(
inputs,
network,
Expand Down

0 comments on commit 721787f

Please sign in to comment.