ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-06-26 18:16:07 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

small edits

Browse Source
This commit is contained in:
revital 2023-10-30 10:24:03 +02:00
parent c59f7e804d
commit 316cef73ba
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
examples/frameworks/huggingface/transformers.ipynb
View File

@ -48,10 +48,10 @@
"# Set to true so ClearML will log the models created by the trainer\n", "# Set to true so ClearML will log the models created by the trainer\n",
"%env CLEARML_LOG_MODEL=True\n", "%env CLEARML_LOG_MODEL=True\n",
"\n", "\n",
"# Set the ClearML task name\n", "# Set the ClearML task name (default \"Trainer\")\n",
"# %env CLEARML_TASK=\n", "# %env CLEARML_TASK=\n",
"\n", "\n",
"# Set task's prkect\n", "# Set task's project (default \"HuggingFace Transformers\")\n",
"# %env CLEARML_PROJECT=" "# %env CLEARML_PROJECT="
], ],
"metadata": { "metadata": {
@ -100,8 +100,7 @@
"id": "LG-TqpPf9_Tn" "id": "LG-TqpPf9_Tn"
}, },
"source": [ "source": [
"Create a trainer, in which you'll typically pass the following parameters to [Trainer](https://huggingface.co/docs/transformers/main/en/main_classes/\n", "Create a trainer, in which you'll typically pass the following parameters to [Trainer](https://huggingface.co/docs/transformers/main/en/main_classes/trainer#transformers.Trainer):\n",
"trainer#transformers.Trainer):\n",
"\n", "\n",
"1. A [PreTrainedModel](https://huggingface.co/docs/transformers/main/en/main_classes/model#transformers.PreTrainedModel) or a [`torch.nn.Module`](https://pytorch.org/docs/stable/nn.html#torch.nn.Module):\n", "1. A [PreTrainedModel](https://huggingface.co/docs/transformers/main/en/main_classes/model#transformers.PreTrainedModel) or a [`torch.nn.Module`](https://pytorch.org/docs/stable/nn.html#torch.nn.Module):\n",
"\n" "\n"