search

NVlabs / Instance_Insertion

Other
87 stars 10 forks source link

Multi-GPUs Traning #6

Open wang-zm18 opened 4 years ago

wang-zm18 commented 4 years ago

I have tried to add "torch.nn.DataParallel" to every model, and modifies the batch_size. However, "nvidia-smi" still shows only one gpu is using. I cannot figure out what's wrong. Thank you in advance!

wang-zm18 commented 4 years ago

Also, after adding "torch.nn.DataParallel" and changing the batchsize, the code cannot run as we expected. The state is shown in the following pictures. error

abhigoku10 commented 2 years ago

@wang-zm18 hi where you able to make this code work ?