Skip to content

Commit

Permalink
typo fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
rasbt committed Aug 8, 2023
1 parent 98dd517 commit 313482f
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -799,7 +799,7 @@
" self.log(\"train_loss\", loss)\n",
"\n",
" predicted_labels = corn_label_from_logits(outputs[\"logits\"])\n",
" self.test_mae(predicted_labels, batch[\"LABEL_COLUMN_NAME\"])\n",
" self.train_mae(predicted_labels, batch[\"LABEL_COLUMN_NAME\"])\n",
" self.log(\"train_mae\", self.train_mae, prog_bar=True)\n",
" \n",
" return loss # this is passed to the optimizer for training\n",
Expand Down Expand Up @@ -1214,7 +1214,7 @@
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
"version": "3.10.10"
"version": "3.10.6"
}
},
"nbformat": 4,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -796,7 +796,7 @@
" self.log(\"train_loss\", outputs[\"loss\"])\n",
"\n",
" predicted_labels = torch.argmax(outputs[\"logits\"], 1)\n",
" self.test_mae(predicted_labels, batch[\"LABEL_COLUMN_NAME\"])\n",
" self.train_mae(predicted_labels, batch[\"LABEL_COLUMN_NAME\"])\n",
" self.log(\"train_mae\", self.train_mae, prog_bar=True)\n",
" \n",
" return outputs[\"loss\"] # this is passed to the optimizer for training\n",
Expand Down Expand Up @@ -1209,7 +1209,7 @@
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
"version": "3.10.10"
"version": "3.10.6"
}
},
"nbformat": 4,
Expand Down

0 comments on commit 313482f

Please sign in to comment.