Hi,
I want to train my model on two gpus and I have set my model to:
model = torch.nn.DataParallel(model).cuda()

However, I notice that only gpu 0 is running and gpu 1 is not doing any work. Did I missed anything?
Best,
R