Skip to content

Commit

Permalink
Added comment to docs regarding ToTensor transform (apache#12186)
Browse files Browse the repository at this point in the history
* Corrections to profiling tutorial

Corrected a race condition with stopping profiling. Added mx.nd.waitall to ensure all operations have completed, including GPU operations that might otherwise be missing.

Also added alternative code for context selection GPU vs CPU, that had error before on machines with nvidia-smi.

* Adding note in documentation regarding ToTensor transform.

* Removed line to fix lint error.

* Change to start CI build
  • Loading branch information
thomelane authored and anirudh2290 committed Sep 19, 2018
1 parent 94a9069 commit 9a6289b
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 1 deletion.
2 changes: 1 addition & 1 deletion docs/api/python/gluon/data.md
Original file line number Diff line number Diff line change
Expand Up @@ -76,7 +76,7 @@ In the rest of this document, we list routines provided by the `gluon.data` pack
```

Transforms can be used to augment input data during training. You
can compose multiple transforms sequentially, for example:
can compose multiple transforms sequentially (taking note of which functions should be applied before and after `ToTensor`).

```python
from mxnet.gluon.data.vision import MNIST, transforms
Expand Down
1 change: 1 addition & 0 deletions python/mxnet/gluon/data/vision/transforms.py
Original file line number Diff line number Diff line change
Expand Up @@ -245,6 +245,7 @@ def forward(self, x):

class Resize(Block):
"""Resize an image to the given size.
Should be applied before `mxnet.gluon.data.vision.transforms.ToTensor`.
Parameters
----------
Expand Down

0 comments on commit 9a6289b

Please sign in to comment.