Skip to content

Commit

Permalink
add comment note about where mps fails in img2pose
Browse files Browse the repository at this point in the history
  • Loading branch information
ejolly committed Apr 6, 2024
1 parent f0fb31b commit 65ad854
Showing 1 changed file with 6 additions and 0 deletions.
6 changes: 6 additions & 0 deletions feat/facepose_detectors/img2pose/img2pose_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -183,6 +183,12 @@ def predict(self, img, border_size=0, scale=1.0, euler=True):
dict: A dictionary of bboxes and poses
"""
# For device='mps'
# Uncommenting this line at least gets img2pose running but errors with
# Error: command buffer exited with error status.
# The Metal Performance Shaders operations encoded on it may not have completed.

# img = img.to(self.device)

# Obtain prediction
pred = self.model.predict([img])[0]
Expand Down

0 comments on commit 65ad854

Please sign in to comment.