search

AIPHES / emnlp19-moverscore

MoverScore: Text Generation Evaluating with Contextualized Embeddings and Earth Mover Distance
MIT License
197 stars 34 forks source link

Update many #18

Closed mymusise closed 3 years ago

mymusise commented 3 years ago
  1. update huggingface/transformers api
  2. add MOVERSCORE_MODEL environment variable to select different bert model. example: 
    
import os 
os.environ['MOVERSCORE_MODEL'] = "albert-base-v2"

from moverscore_v2 import get_idf_dict idf_dict_hyp = get_idf_dict(translations)


3. add `moverscore_v2` to setup.py
4. suggest use `CUDA_VISIBLE_DEVICES` instead of `device` parameter

Hope it help!