when train gap task ,I got this error ,is it a bug?

vid-koci / bert-commonsense

Code for papers "A Surprisingly Robust Trick for Winograd Schema Challenge" and "WikiCREM: A Large Unsupervised Corpus for Coreference Resolution"

70 stars 13 forks source link

when train gap task ,I got this error ,is it a bug? #8

Open houliangxue opened 2 years ago

houliangxue commented 2 years ago

when I run: python /main.py --task_name gap --do_eval --do_train --eval_batch_size 10 --data_dir "autodl-tmp/bert-commonsense-master/data/" --bert_model bert-large-uncased --max_seq_length 128 --train_batch_size 16 --alpha_param 20 --beta_param 0.2 --learning_rate 5.0e-6 --num_train_epochs 1.0 --output_dir autodl-tmp/bert-commonsense-master/model_output/ I got error, what's the problem ,please

vid-koci commented 2 years ago

Hi, this error seems to be connected to CUDA/PyTorch and Nvidia driver error. Unfortunately, the code is not supported for more recent versions of PyTorch/CUDA, making it incompatible with newer Nvidia drivers. You can try running it with newer versions, see this issue: https://github.com/vid-koci/bert-commonsense/issues/6