如何多卡训练？

[bosslv]

怎么样修改才能多卡训练？

[xiongxyowo]

You can simply use model = nn.DataParallel(model):
https://pytorch.org/tutorials/beginner/blitz/data_parallel_tutorial.html