Fix GPU selection - Githubissues

facebookresearch / VMZ

VMZ: Model Zoo for Video Modeling

Apache License 2.0

1.04k stars 156 forks source link

Fix GPU selection #64

Closed ohtake closed 5 years ago

ohtake commented 5 years ago

Running extract_features.py with --gpus=0,2 fails. The error is:

  result = C.fetch_blob(StringifyBlobName(name))
RuntimeError: [enforce fail at pybind_state.cc:183] ws->HasBlob(name). Can't find blob: gpu_1/softmax

extract_features.py has an assumption that there is no gaps in GPU identifiers.

dutran commented 5 years ago

@ohtake: Thanks for the fix. It looks good to me.