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
If a cuda tensor is passed it will throw an error as torch.randn will generate a cpu tensor and thus both won't be same type. Either throw an explicit warning or check for type of label_img and convert rand tensor accordingly.
I solved it by converting passed tensor to cpu for now.
The text was updated successfully, but these errors were encountered:
apsdehal
changed the title
Cat issue in add_embedding
Cat issue in add_embedding with cuda tensor
Dec 14, 2017
In line number: https://github.com/lanpa/tensorboard-pytorch/blob/master/tensorboardX/embedding.py#L20
If a cuda tensor is passed it will throw an error as torch.randn will generate a cpu tensor and thus both won't be same type. Either throw an explicit warning or check for type of
label_img
and convert rand tensor accordingly.I solved it by converting passed tensor to cpu for now.
The text was updated successfully, but these errors were encountered: