You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Traceback (most recent call last):
File "train2.py", line 217, in <module>
train(opt, logger)
File "train2.py", line 88, in train
target_lpsoft, target_image_full, target_weak_params)
File "/media/ailab/data/yy/ProDA/models/adaptation_modelv2.py", line 209, in step
weights = self.get_prototype_weight(ema_out['feat'], target_weak_params=target_weak_params)
File "/media/ailab/data/yy/ProDA/models/adaptation_modelv2.py", line 350, in get_prototype_weight
feat_proto_distance = self.feat_prototype_distance(feat)
File "/media/ailab/data/yy/ProDA/models/adaptation_modelv2.py", line 345, in feat_prototype_distance
feat_proto_distance[:, i, :, :] = torch.norm(self.objective_vectors[i].reshape(-1,1,1).expand(-1, H, W) - feat, 2, dim=1,)
RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!
I only want to use two gpu, So I change the code only in here
when I run tran.py
I only want to use two gpu, So I change the code only in here
The text was updated successfully, but these errors were encountered: