multi-gpu running

[mmuguang]

Hi! How to modify the code to adapt multi-gpu running! I simple set the device=[0,1],but it didn't work.

[JanekEbb]

Hi, you need to set trainer.virtual_minibatch_size=#gpus (=2 in your case)