Skip to content

fix: the readme and train codes do not match#2

Open
LongQIByte wants to merge 1 commit intohy-struggle:mainfrom
LongQIByte:main
Open

fix: the readme and train codes do not match#2
LongQIByte wants to merge 1 commit intohy-struggle:mainfrom
LongQIByte:main

Conversation

@LongQIByte
Copy link
Copy Markdown

Download [BERT-Base-Cased](https://huggingface.co/bert-base-cased/tree/main) which contains `pytroch_model.bin`, `vocab.txt` and `config.json`. Put these under `./pretrain_models/`.

👆🏻👆🏻👆🏻Above will display an error, do not have this folder(/pretrain_models/bert_base_cased/)

# in train.py

bert_config = BertConfig.from_json_file(os.path.join(params.bert_model_dir, 'bert_config.json'))

👆🏻👆🏻👆🏻Above will display an error, can't find bert_config.json.In fact just download should be config.json

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant