Can this code be run with multiple GPUs?

[leaf-ygq]

Hi @EmilyAlsentzer , I want to know if the code can be run with multiple GPUs. I change the
parameter of trainer in train_config.py.

But there was something wrong after I finished it.

I'm sure that there is no bug when I use only one GPU. I really hope that you could help me deal with it! Please feel free to concat me.