PyCIL
PyCIL copied to clipboard
When i run codes on GPU1, it seems that .cuda() will put data on default GPU and then bugs occurred.
When i run codes on GPU1, it seems that .cuda() will put data on default GPU and then bugs occurred.
File "/home/ubuntu/lzc/PyCIL/utils/inc_net.py", line 605, in forward
fc_weight = torch.cat([self.old_fc.weight,torch.zeros((new_task_size,self.feature_dim-self.out_dim)).cuda()],dim=0)
RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:1 and cuda:0! (when checking argument for argument tensors in method wrapper_cat)
这是来自QQ邮箱的假期自动回复邮件。 您好,我最近正在休假中,无法亲自回复您的邮件。我将在假期结束后,尽快给您回复。
Thanks for informing us. The bug will be fixed in the upcoming version.