question about multiple GPUs training

[leefly072]

Thank you very much for sharing. I have some mistakes when I wanted to do GPU training in parallel. It seems that the code can only be trained on one GPU. There is no " torch.nn.DataParallel" in the code. could it also can train in multiple GPUs?