Skip to content
This repository has been archived by the owner on Dec 21, 2023. It is now read-only.

Fix single-image classifier inference on multi-gpu #417

Conversation

gustavla
Copy link
Collaborator

@gustavla gustavla commented Mar 31, 2018

MXNet fatally crashes when you give zero batches to a GPU (I really don't know why they chose to make this a fatal error!), so we have to
limit the contexts based on the batch size. I believe OD and AC already do this.

This bug is caught by the recent test test_single_image when gpus>=2.

MXNet fatally crashes when you give zero batches to a GPU, so we have to
limit the contexts based on the batch size. I believe OD and AC already do this.
@gustavla gustavla merged commit e11ac89 into apple:master Apr 2, 2018
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants