asitstands commented on issue #11268: A binary RBM example URL: https://github.com/apache/incubator-mxnet/pull/11268#issuecomment-408650170 @yifeim It was my fault that the update was late. I forgot to push the update and went on a vacation :grin:. I pushed the ais implementation and some tweaks yesterday. I found that the gluon implementation is much slower than the module implementation. nvidia-smi reports about 60% of GPU utilization for the module implementation but only 35% for the gluon implementation. If I replace gluon's `DataLoader` with `NDArrayIter`, the gluon implementation's performance becomes comparable to the module implementation. I'm not sure that this performace issue is unavoidable with `DataLoader` or I have made some mistakes. Would you please check this performance issue?
---------------------------------------------------------------- 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: [email protected] With regards, Apache Git Services
