I am reading a code about CMC downloaded from github, and it can\'t work on vscode.
Code:
if torch.cuda.is_available(): index = index.cuda(asyn