We present code for training models described in the paper, as well as pre-trained models. We tested scripts using an NVIDIA A40 gpu in a single setting. In multi-gpu settings, we found that there is ...
The package directly uses the "six" Python module but does not list it as a dependency in its setup file which can cause failures in some Python environments when six isn't already installed.