... y_test.to(device) y_test = y_test.squeeze(1) outputs = model(x_test) _, ... y_test).sum().item() detached_pred = predicted.detach().cpu().numpy() ...
確定! 回上一頁