run_biorex_exp.sh doesn't seem to use GPU

[dongheechoi]

I tried to run the script but I don't think it is using a GPU. I stopped the code before this line (https://github.com/ncbi/BioREx/blob/0031c52d2dd0d7fb8e3b84e256d8fbf73c4bd463/src/tf_wrapper.py#L199C28-L200C1), and check it whether using GPU or not, but I got this.

In [16]: self.model.layers[0].variables[0].device
Out[16]: '/job:localhost/replica:0/task:0/device:CPU:0'

Can you kindly check the script or runfile to use GPU to train?

And I think it is better to change the static gpu variable in the script into dynamic one as you put in the other script. I also put that in the pull request(#4)

BioREx/scripts/run_biorex_exp.sh

Line 3 in 0031c52

cuda_visible_devices=0

[ptlai]

Hi @dongheechoi ,

The program can run on GPU as below.

Yes, you should be able to change cuda_visible_devices to the visible GPUs.