Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

can i trained it with cpu? #124

Open
shadow12138 opened this issue Dec 19, 2019 · 1 comment
Open

can i trained it with cpu? #124

shadow12138 opened this issue Dec 19, 2019 · 1 comment

Comments

@shadow12138
Copy link

how should i config to train it with cpu?
the default option for "gpus" is 0,
and the gpu environment is set in variable list "ctx"
when i set ctx to [], it raises index out of bound error

@ijkguo @li-haoran

@shadow12138
Copy link
Author

i changed the Module's context parameter to mx.cpu()
it started to run now, but it seemed like it's stucked.
with a warning in bbox.py:

RuntimeWarning: invalid value encountered in log
targets_dw = np.log(gt_widths / ex_widths) / box_stds[2]

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant