search

wzhouad / ATLOP

Source code for paper "Document-Level Relation Extraction with Adaptive Thresholding and Localized Context Pooling", AAAI 2021
197 stars 39 forks source link

Multi-gpu code #6

Closed otpas007 closed 3 years ago

otpas007 commented 3 years ago

Can you please provide the multi-gpu version of the code?

wzhouad commented 3 years ago

You can use larger gradient accumulation steps and a smaller batch size if you have an OOM issue. I don't think multi-gpu is necessary.