Hi,
I would like to apply your model on my own dataset, so I generated the graphs following the instructions. I hit a wall however when I try to generate the predictions on CPU. I set the use_gpu parameter in the config file, but it seems like a couple of parameters are still on the GPU.
Also, I had to switch to RobertaTokenizer instead of BertTokenizer, but I don't think this would cause my problem.
Hi, I would like to apply your model on my own dataset, so I generated the graphs following the instructions. I hit a wall however when I try to generate the predictions on CPU. I set the use_gpu parameter in the config file, but it seems like a couple of parameters are still on the GPU. Also, I had to switch to RobertaTokenizer instead of BertTokenizer, but I don't think this would cause my problem.