Backpropagation error

torch.package / torch::deploy
3

return self.criterion(y, finalTrueDist.float()) works. Thank you!