Skip to content

Commit f05d6aa

Browse files
authored
update EnsureTyped usage (#1405)
Fixes #1404 ### Checks <!--- Put an `x` in all the boxes that apply, and remove the not applicable items --> - [ ] Avoid including large-size files in the PR. - [ ] Clean up long text outputs from code cells in the notebook. - [ ] For security purposes, please check the contents and remove any sensitive info such as user names and private key. - [ ] Ensure (1) hyperlinks and markdown anchors are working (2) use relative paths for tutorial repo files (3) put figure and graphs in the `./figure` folder - [ ] Notebook runs automatically `./runner.sh -t <path to .ipynb file>` Signed-off-by: Wenqi Li <[email protected]>
1 parent e544dd6 commit f05d6aa

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

acceleration/fast_model_training_guide.md

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -270,14 +270,13 @@ For example:
270270
```py
271271
train_transforms = [
272272
LoadImaged(...),
273-
AddChanneld(...),
273+
EnsureChannelFirstd(...),
274274
Orientationd(...),
275275
Spacingd(...),
276276
ScaleIntensityRanged(...),
277277
CropForegroundd(...),
278278
FgBgToIndicesd(...),
279-
EnsureTyped(..., data_type="tensor"),
280-
ToDeviced(..., device="cuda:0"),
279+
EnsureTyped(..., device="cuda:0"),
281280
RandCropByPosNegLabeld(...),
282281
]
283282
dataset = CacheDataset(..., transform=train_trans)

0 commit comments

Comments
 (0)