nn.parallel.DataParallel https://discuss.pytorch.org/t/is-the-loss-function-paralleled-when-using-dataparallel/3346
nn.parallel.DataParallel