bvanaken/clinical-outcome-prediction

[Main Readme] Prediction task Training Argument Name should be changed

Closed this issue · 1 comments

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!

Thanks @bhadreshpsavani! You are right, the parameter name was outdated. I fixed it in the README.