fix bug of word_freq - Githubissues

Hzfinfdu / Diffusion-BERT

ACL'2023: DiffusionBERT: Improving Generative Masked Language Models with Diffusion Models

Apache License 2.0

292 stars 25 forks source link

fix bug of word_freq #13

Closed lpyhdzx closed 1 year ago

lpyhdzx commented 1 year ago

Hi, This is great work! I met a problem when running word_freq.py. I guess line #18 in file word_freq.py should be changed as follows. I hope this modification will be useful. #original for iid in train_data['input_ids']: # modified for iid in data['input_ids']: