albert_pytorch
albert_pytorch copied to clipboard
这个断言可以修改一下,前面内循环判断。 if len(masked_token_labels) >= num_to_mask:
我也遇到了这个情况,按照作者的原始意图,在下面的循环中加上 即可
if len(masked_token_labels) >= num_to_mask:
break
是加在内循环吧,我看着外循环已经有了