torchscale/examples/fairseq/criterions/__init__.py