CollaboNet icon indicating copy to clipboard operation
CollaboNet copied to clipboard

Model runs on CPU instead of GPU

Open youssef-snoussi opened this issue 4 years ago • 0 comments

Hello, I tried to launch the single STM learning using the command provided in the Readme file

python run.py --ncbi --jnlpba --bc5_chem --bc5_disease --bc4 --bc2 --lr_pump --lr_decay 0.05

however while running the training I had the suspicion that it runs on CPU instead of the GPU, here are some screenshots that shows the CPU and the GPU usage during the training. Screenshot from 2021-04-04 14-56-53 Screenshot from 2021-04-04 19-28-13

Is this a normal thing or should I change something?

youssef-snoussi avatar Apr 03 '21 00:04 youssef-snoussi