Skip to content

Fix Mulit-GPU problem#2

Open
rsk25 wants to merge 7 commits intoMiuLab:masterfrom
rsk25:master
Open

Fix Mulit-GPU problem#2
rsk25 wants to merge 7 commits intoMiuLab:masterfrom
rsk25:master

Conversation

@rsk25
Copy link

@rsk25 rsk25 commented May 12, 2021

The original code raised errors when I ran it with torch=>1.4.0 and multi-GPU. It seems like you guys forgot to add 'info_loss.mean()' when using multi-GPU in run_squad.py line 998.

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

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant