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

Representation Metrics in DDP mode #149

Open
PososikTeam opened this issue Oct 10, 2022 · 0 comments
Open

Representation Metrics in DDP mode #149

PososikTeam opened this issue Oct 10, 2022 · 0 comments
Assignees
Labels
bug Something isn't working

Comments

@PososikTeam
Copy link
Collaborator

Describe the bug
In DDP mode representation metrics doesn't work. Due to in update function gpu embedding convert to cpu embedding and after that in compute method in DDP mode called all_gather() which work only with gpu tensors.

@PososikTeam PososikTeam added the bug Something isn't working label Oct 10, 2022
@PososikTeam PososikTeam self-assigned this Oct 10, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug Something isn't working
Projects
None yet
Development

No branches or pull requests

1 participant