Skip to content

Commit

Permalink
Browse files Browse the repository at this point in the history
More fixes
  • Loading branch information
Simon Kozlov authored and Simon Kozlov committed Apr 4, 2019
1 parent 08c705e commit c74d819
Showing 1 changed file with 3 additions and 4 deletions.
7 changes: 3 additions & 4 deletions assignments/assignment2/PyTorch.ipynb
Expand Up @@ -236,8 +236,7 @@
" model.eval() # Evaluation mode\n",
" # TODO: Implement the inference of the model on all of the batches from loader,\n",
" # and compute the overall accuracy.\n",
" # Hint: torch doesn't have a dedicated argmax function,\n",
" # but you can use torch.max instead (see the documentation).\n",
" # Hint: PyTorch has the argmax function!\n",
" \n",
" raise Exception(\"Not implemented\")\n",
" \n",
Expand Down Expand Up @@ -370,8 +369,8 @@
" Computes predictions and ground truth labels for the indices of the dataset\n",
" \n",
" Returns: \n",
" predictions: np array of booleans of model predictions\n",
" grount_truth: np array of boolean of actual labels of the dataset\n",
" predictions: np array of ints - model predictions\n",
" grount_truth: np array of ints - actual labels of the dataset\n",
" \"\"\"\n",
" model.eval() # Evaluation mode\n",
" \n",
Expand Down

0 comments on commit c74d819

Please sign in to comment.