CollaboNet
CollaboNet copied to clipboard
Model runs on CPU instead of GPU
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.

Is this a normal thing or should I change something?