Skip to content

Commit 5bd4c2c

Browse files
committed
Add cuda device arg to inpainting example
1 parent edf44fb commit 5bd4c2c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

Diff for: README.md

+1-1
Original file line numberDiff line numberDiff line change
@@ -197,7 +197,7 @@ segmenter.infer_single(
197197
from brats import Inpainter
198198
from brats.constants import InpaintingAlgorithms
199199

200-
inpainter = Inpainter(algorithm=InpaintingAlgorithms.BraTS24_1)
200+
inpainter = Inpainter(algorithm=InpaintingAlgorithms.BraTS24_1, cuda_devices="0")
201201
inpainter.infer_single(
202202
t1n="path/to/voided_t1n.nii.gz",
203203
mask="path/to/mask.nii.gz",

0 commit comments

Comments
 (0)