Hanjun Goo

icon indicating an email [email protected]

icon location Seoul, Korea icon location Seoul National Univ. Phd candidate

Results 1 comments of Hanjun Goo

Failing to reproduce the code

line 137, 140 must be pred_items.scatter_(dim=1, index=test_indices,src=torch.ones_like(test_indices).float().cuda()) topk_preds.scatter_(dim=1, index=test_indices[:, :k], src=torch.ones_like(test_indices[:, :k]).float().cuda())