Help! How to set torch.nn.DataParallel()?

Hi, Guys.

I am new in deep learning area. Now I am reproducing a paper’s codes. since they use several GPUs, there is a command “torch.nn.DataParallel(model, device_ids= args.gpus).cuda()” in codes. But I only have one GPU, what
should I change this code to match up my GPU?

Thank you!

model.cuda() is enough.