AttributeError: 'Tensor' object has no attribute 'T'

facebookresearch / moco

PyTorch implementation of MoCo: https://arxiv.org/abs/1911.05722

MIT License

4.83k stars 794 forks source link

AttributeError: 'Tensor' object has no attribute 'T' #57

Closed chinglamchoi closed 4 years ago

chinglamchoi commented 4 years ago

Got the above error when running moco/builder.py line 63: self.queue[:, ptr:ptr + batch_size] = keys.T. Does this .T refer to the softmax temperature?

ppwwyyxx commented 4 years ago

see https://pytorch.org/docs/stable/tensors.html#torch.Tensor.T

chinglamchoi commented 4 years ago

Ah thanks, version mismatch problem (I'm on 1.0.0).