BadBubble

Results 5 comments of BadBubble

@licheng-pro 如果你数据中的label已经转成了label id: ```python def get_labels(self): return [str(i) for i in range(n_classes)] ```

@licheng-pro 需要return是lable id,而不是label! try this: ```python def get_labels(self, label_path): fo = open(label_path, 'r', encoding='utf-8') lines = fo.readlines() label_to_id ={} for i, line in enumerate(lines): label_to_id[line.strip()] = i print(label_to_id) fo.close() return...

@licheng-pro 刚去看了下代码确实是:stuck_out_tongue:,不过我做2k个类别没有任何问题。 ```RuntimeError: Assertion `cur_target >= 0 && cur_target < n_classes' failed. at /pytorch/aten/src/THNN/generic/ClassNLLCriterion.c:93``` 我个人感觉这个错误肯定是label出了问题,比如`label_path`中的数据有重复, 你可以 ```return list(set(label_list))``` 试一下

你用管理员身份运行CMD试试

@liubo12 @flybird101 - [bert-base-uncased](https://s3.amazonaws.com/models.huggingface.co/bert/bert-base-uncased.tar.gz) - [bert-large-uncased](https://s3.amazonaws.com/models.huggingface.co/bert/bert-large-uncased.tar.gz) - [bert-base-cased](https://s3.amazonaws.com/models.huggingface.co/bert/bert-base-cased.tar.gz) - [bert-large-cased](https://s3.amazonaws.com/models.huggingface.co/bert/bert-large-cased.tar.gz) - [bert-base-multilingual-uncased](https://s3.amazonaws.com/models.huggingface.co/bert/bert-base-multilingual-uncased.tar.gz) - [bert-base-multilingual-cased](https://s3.amazonaws.com/models.huggingface.co/bert/bert-base-multilingual-cased.tar.gz) - [bert-base-chinese](https://s3.amazonaws.com/models.huggingface.co/bert/bert-base-chinese.tar.gz)