search

neulab / awesome-align

A neural word aligner based on multilingual BERT
https://arxiv.org/abs/2101.08231
BSD 3-Clause "New" or "Revised" License
325 stars 47 forks source link

Multiple GPUs #10

Closed BramVanroy closed 3 years ago

BramVanroy commented 3 years ago

When using multiple GPUs, the train script does not work.

AttributeError: 'DataParallel' object has no attribute 'get_aligned_word'