IrvingShu commented on issue #9920: Asynchronous Issue on CustomOP and mxnet.image.ImageDetIter
URL: https://github.com/apache/incubator-mxnet/issues/9920#issuecomment-382997920
@wkcn thanks for sharing your ways. I added ctx = context.cpu(), however, the new problem
happens:
/usr/local/lib/python2.7/dist-packages/mxnet/module/base_module.py:466: UserWarning: Optimizer
created manually outside Module but rescale_grad is not normalized to 1.0/batch_size/num_workers
(0.25 vs. 0.0078125). Is this intended?
optimizer_params=optimizer_params)
call reset()
terminate called after throwing an instance of 'dmlc::Error'
what(): [14:37:42] src/engine/./threaded_engine.h:359: [14:37:42] src/ndarray/ndarray_function.cu:43:
Check failed: to->type_flag_ == from.type_flag_ (0 vs. 3) Source and target must have the
same data type when copying across devices.
Did you solve it ?
----------------------------------------------------------------
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:
users@infra.apache.org
With regards,
Apache Git Services
|