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
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'