[Main Readme] Prediction task Training Argument Name should be changed
Closed this issue · 1 comments
bhadreshpsavani commented
Hi @bvanaken,
I was trying to reproduce things and during downstream training, the below code was giving an error
python doc_classification.py \
--task_config {PATH_TO_TASK_CONFIG.yaml} \ # required
--model_name {PATH_TO_MODEL_OR_TRANSFORMERS_MODEL_HUB_NAME} \ # required
--cache_dir {CACHE_DIR} \ # required
when I changed it to this
python doc_classification.py \
--task_config {PATH_TO_TASK_CONFIG.yaml} \ # required
--model_name_or_path {PATH_TO_MODEL_OR_TRANSFORMERS_MODEL_HUB_NAME} \ # required
--cache_dir {CACHE_DIR} \ # required
I mean model_name
to model_name_or_path
it worked!
bvanaken commented
Thanks @bhadreshpsavani! You are right, the parameter name was outdated. I fixed it in the README.