Use PyTorch with DC/OS Data Science Engine. ... get the index of the max log-probability correct += pred.eq(target.view_as(pred)).sum().item() test_loss ...
確定! 回上一頁