INK-USC / CrossFit

Code for paper "CrossFit :weight_lifting:: A Few-shot Learning Challenge for Cross-task Generalization in NLP" (https://arxiv.org/abs/2104.08835)
105 stars 6 forks source link

ModuleNotFoundError: No module named 'transformers.models' #4

Closed harkiratbehl closed 2 years ago

harkiratbehl commented 2 years ago

python tune_hps_singletask.py --task_dir data/boolq/ --do_train --do_predict --learning_rate_list 1e-5 2e-5 5e-5 --bsz_list 2 4 8 --total_steps 1000 --eval_period 100 --warmup_steps 100 --model facebook/bart-base --output_dir models/singletask-boolq --predict_batch_size 32 Traceback (most recent call last): File "tune_hps_singletask.py", line 30, in from run_singletask import run File "/home/harkirat/research/2022/crossfit_polytropon/run_singletask.py", line 11, in from bart import MyBart, MySkilledBart File "/home/harkirat/research/2022/crossfit_polytropon/bart.py", line 17, in from transformers.models.bart.modeling_bart import shift_tokens_right ModuleNotFoundError: No module named 'transformers.models'