CogComp / ZeroShotWiki

Towards Open Domain Topic Classification
MIT License
3 stars 0 forks source link

getting error message when training #1

Closed anjanatiha closed 2 years ago

anjanatiha commented 2 years ago

--:/shared/tiha/code_review/ZeroShotWiki> CUDA_VISIBLE_DEVICES=1 python src/train_wiki_entailment.py --data_file data/data/wikipedia/tokenized_wiki.txt --cate_file data/data/wikipedia/page2content_cate.json --save_dir "model2" 07/26/2022 16:40:32 - WARNING - transformers.modeling_utils - Some weights of the model checkpoint at bert-base-uncased were not used when initializing BertForSequenceClassification: ['cls.predictions.bias', 'cls.predictions.transform.dense.weight', 'cls.predictions.transform.dense.bias', 'cls.predictions.decoder.weight', 'cls.seq_relationship.weight', 'cls.seq_relationship.bias', 'cls.predictions.transform.LayerNorm.weight', 'cls.predictions.transform.LayerNorm.bias']

dht2345 commented 2 years ago

Resolved. https://github.com/CogComp/ZeroShotWiki/commit/74149d3842e07270b980f2bda570f3934de28481