RuntimeError: CUDA error: invalid device function

dandelin / ViLT

Code for the ICML 2021 (long talk) paper: "ViLT: Vision-and-Language Transformer Without Convolution or Region Supervision"

Apache License 2.0

1.36k stars 209 forks source link

RuntimeError: CUDA error: invalid device function #72

Open lonestar234028 opened 1 year ago

lonestar234028 commented 1 year ago

Hi Expert: I have downloaded the images and stored them into arrow, then when I run the script of NLVR2, there are some errors, can you give me some guidance? I have checked a lot of times, and do not know how to resolve it.

python run.py with data_root={arrowr_root} num_gpus=1 num_nodes=1 per_gpu_batchsize=64 task_finetune_nlvr2_randaug test_only=True precision=32 load_path={mlvr2ckpt}

lonestar234028 commented 1 year ago

Gitspike commented 1 year ago

have you solved this problem?I met it too