seba-1511/dist_tuto.pth

Whether we maintain two same models in two process

Opened this issue · 1 comments

model = Net()

Whether it isn't useful in any real practice? I think eventually we only need one model
How I can know which GPU place each data replicas and each model?

we need use some codes like:

 device = torch.device("cuda:{}".format(rank))
  model = Net().to(device)
 data, target = data.to(device), target.to(device)