variance_reduced_neural_networks icon indicating copy to clipboard operation
variance_reduced_neural_networks copied to clipboard

Published 20 hours ago • verified publisher icon kilianFatras

FIX torch.data[0]=>torch.item()

Open limaries30 opened this issue 4 years ago • 0 comments

self.partial_grad(inputs, labels, loss_function).data[0] causes error.

So I changed it to self.partial_grad(inputs, labels, loss_function).item()

limaries30 avatar Dec 30 '20 08:12 limaries30