Koen Dejonghe created MXNET-428:
-----------------------------------
Summary: Switch data type for NDArray creation
Key: MXNET-428
URL: https://issues.apache.org/jira/browse/MXNET-428
Project: Apache MXNet
Issue Type: Improvement
Reporter: Koen Dejonghe
[Scala]
It should be possible to swap the data type from float to double for creation
of NDArrays, in order to perform gradient checks.
For more information on gradient checks, see
http://cs231n.github.io/neural-networks-3/#gradcheck
As an idea, in ND4J (https://nd4j.org/) it is possible to globally set the data
type by executing:
Nd4j.setDataType(DataBuffer.Type.DOUBLE)
Subsequent nd-arrays will be of type Double.
--
This message was sent by Atlassian JIRA
(v7.6.3#76005)
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
