.cpu().detach().numpy() vs .data.cpu().numpy()

[pytorching]

Thank you very much. That answers my question exactly.
BTW, I don’t know if it’s allowed or forbid. May I ask you to look at another question I post?

nn.DataParallel: TypeError: expected sequence object with len >= 0 or a single integer

In my forward function: def __call__(self, train=True): if train: predicted = self.forward(...) loss = .... return loss # return a single value that's fine # loss.size() = the number of my GPUs. else: predicted = self.forward(...) return predicted # expected sequence object with len >= 0 or a single integer # In validation step I want to return the whole predict labels for other purpose # predicted with shape [16, 1] on eac…