gpu memory usage

[Cumberbatch08]

when i use nvidia-smi to watch gpu usage,

I have three gpus, but it seems like only one gpu is used?

[QuantumChamploo]

the default gpu value on these examples is set at one, unless you explicitly declare otherwise. Im not super familiar with these packages, but thats the issue I had when I ran this.

Hope that helps, cheers

[Cumberbatch08]

the code defines the number of gpu, and in line 134, the for loop uses the gpus.

[Cumberbatch08]

when I run the code, my code did'nt run the train_op, so this code works, and load the model to predict also works.
many thaks!