Skip to content

Commit c123d58

Browse files
committed
Move new detector to device
1 parent fe62a48 commit c123d58

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

diffdrr/drr.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -169,7 +169,7 @@ def set_intrinsics(
169169
y0 if y0 is not None else self.detector.y0,
170170
n_subsample=self.detector.n_subsample,
171171
reverse_x_axis=self.detector.reverse_x_axis,
172-
)
172+
).to(self.volume)
173173

174174
# %% ../notebooks/api/00_drr.ipynb 14
175175
from .utils import convert

notebooks/api/00_drr.ipynb

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -303,7 +303,7 @@
303303
" y0 if y0 is not None else self.detector.y0,\n",
304304
" n_subsample=self.detector.n_subsample,\n",
305305
" reverse_x_axis=self.detector.reverse_x_axis,\n",
306-
" )"
306+
" ).to(self.volume)"
307307
]
308308
},
309309
{

0 commit comments

Comments
 (0)