[GitHub] rahul003 opened a new issue #9774: mx.io.ImageRecordIter does not respect dtype argument
rahul003 opened a new issue #9774: mx.io.ImageRecordIter does not respect dtype argument URL: https://github.com/apache/incubator-mxnet/issues/9774 ## Description mx.io.ImageRecordIter or src/io/iter_image_recordio_2.cc doesn't respect dtype parameter taken. It is designed to only work with float32 because of instantiating the class with real_t dtype. (in src/io/iter_image_recordio_2.cc). Can we make it handle fp16 too? This is important for fp16 training. ## Environment info (Required) Mxnet 1.0 Package used: Python ## Error Message: Silently generates fp32 data ## Minimum reproducible example N/A ## Steps to reproduce N/A ## What have you tried to solve it? Can we come up with a better way than to create a new operator passing DType as fp16? @ptrendx This is an automated message from the Apache Git Service. To respond to the message, please log on GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: us...@infra.apache.org With regards, Apache Git Services
[GitHub] rahul003 opened a new issue #9774: mx.io.ImageRecordIter does not respect dtype argument / FP16 performance on Volta
rahul003 opened a new issue #9774: mx.io.ImageRecordIter does not respect dtype argument / FP16 performance on Volta URL: https://github.com/apache/incubator-mxnet/issues/9774 ## Description `mx.io.ImageRecordIter` or `src/io/iter_image_recordio_2.cc` doesn't respect dtype parameter taken. It is designed to only work with float32 because of instantiating the class with real_t dtype. (in src/io/iter_image_recordio_2.cc). Can we make it handle fp16 too? This is important for fp16 training. Also, training in fp16 seems slower than fp32 for some models. ## Environment info (Required) Mxnet 1.0 Package used: Python ## Error Message: Silently generates fp32 data ## Minimum reproducible example N/A ## Steps to reproduce N/A ## What have you tried to solve it? Can we come up with a better way than to create a new operator passing DType as fp16? @ptrendx This is an automated message from the Apache Git Service. To respond to the message, please log on GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: us...@infra.apache.org With regards, Apache Git Services