Skip to content

fixed runtime device error#210

Open
antelopoul wants to merge 1 commit intosalesforce:mainfrom
antelopoul:videoretrieval
Open

fixed runtime device error#210
antelopoul wants to merge 1 commit intosalesforce:mainfrom
antelopoul:videoretrieval

Conversation

@antelopoul
Copy link

encoder_output = video_feats[topk_idx].to(device, non_blocking=True)

RuntimeError: indices should be either on cpu or on the same device as the indexed tensor (cpu)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant