Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
Update fp16 docs: Block.cast is inplace (#15458)
Browse files Browse the repository at this point in the history
Signed-off-by: Serge Panev <[email protected]>
  • Loading branch information
Kh4L authored and szha committed Jul 5, 2019
1 parent 728b8db commit dfe923a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docs/faq/float16.md
Original file line number Diff line number Diff line change
Expand Up @@ -42,7 +42,7 @@ With Gluon API, you need to take care of three things to convert a model to supp
1. Cast Gluon `Block`'s parameters and expected input type to float16 by calling the [cast](https://mxnet.incubator.apache.org/api/python/gluon/gluon.html#mxnet.gluon.Block.cast) method of the `Block` representing the network.

```python
net = net.cast('float16')
net.cast('float16')
```

2. Ensure the data input to the network is of float16 type. If your `DataLoader` or `Iterator` produces output in another datatype, then you would have to cast your data. There are different ways you can do this. The easiest would be to use the [astype](https://mxnet.incubator.apache.org/api/python/ndarray/ndarray.html#mxnet.ndarray.NDArray.astype) method of NDArrays.
Expand Down

0 comments on commit dfe923a

Please sign in to comment.