Skip to content
Merged
Show file tree
Hide file tree
Changes from 3 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 4 additions & 0 deletions tutorials/text/distributed-data-classification/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -29,3 +29,7 @@ For more information about the classifiers, refer to our [Distributed Data Class
| `QualityClassifier` | [quality-classifier-deberta](https://huggingface.co/nvidia/quality-classifier-deberta) |

</div>

## Bring Your Own Classifier

Advanced users may want to integrate their own Hugging Face classifier(s) into NeMo Curator. Broadly, this requires creating a `CompositeStage` consisting of a CPU-based tokenizer stage and a GPU-based model inference stage. Refer to the [Text Classifiers README](https://github.com/NVIDIA-NeMo/Curator/tree/main/nemo_curator/stages/text/classifiers#text-classifiers) for details about how to do this.
Original file line number Diff line number Diff line change
Expand Up @@ -14,9 +14,9 @@
" - Volta™ or higher (compute capability 7.0+)\n",
" - CUDA 12.x\n",
"\n",
"Before running this notebook, see this [Installation Guide](https://docs.nvidia.com/nemo/curator/latest/admin/installation.html#admin-installation) page for instructions on how to install NeMo Curator. Be sure to use an installation method which includes GPU dependencies.\n",
"For more information about the classifiers, refer to our [Distributed Data Classification](https://docs.nvidia.com/nemo/curator/latest/curate-text/process-data/quality-assessment/distributed-classifier.html) documentation page.\n",
"\n",
"For more information about the classifiers, refer to our [Distributed Data Classification](https://docs.nvidia.com/nemo/curator/latest/curate-text/process-data/quality-assessment/distributed-classifier.html) documentation page."
"Before running this notebook, see this [Installation Guide](https://docs.nvidia.com/nemo/curator/latest/admin/installation.html#admin-installation) page for instructions on how to install NeMo Curator. Be sure to use an installation method which includes GPU dependencies (`text_cuda12` or `all`). Check proper installation with:"
]
},
{
Expand All @@ -25,12 +25,50 @@
"metadata": {},
"outputs": [],
"source": [
"# Silence Curator logs via Loguru\n",
"# First, silence Curator logs via Loguru\n",
"import os\n",
"\n",
"os.environ[\"LOGURU_LEVEL\"] = \"ERROR\""
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"import nemo_curator\n",
"\n",
"nemo_curator.__version__ # should be >= 1.0.0"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"We can check that GPUs are available, then check that the `gpustat` dependency was installed:"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"!nvidia-smi"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"import gpustat\n",
"\n",
"gpustat.__version__ # check gpu dependency is installed"
]
},
{
"cell_type": "markdown",
"metadata": {},
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -12,23 +12,61 @@
" - Volta™ or higher (compute capability 7.0+)\n",
" - CUDA 12.x\n",
"\n",
"Before running this notebook, see this [Installation Guide](https://docs.nvidia.com/nemo/curator/latest/admin/installation.html#admin-installation) page for instructions on how to install NeMo Curator. Be sure to use an installation method which includes GPU dependencies.\n",
"For more information about the classifiers, refer to our [Distributed Data Classification](https://docs.nvidia.com/nemo/curator/latest/curate-text/process-data/quality-assessment/distributed-classifier.html) documentation page.\n",
"\n",
"For more information about the classifiers, refer to our [Distributed Data Classification](https://docs.nvidia.com/nemo/curator/latest/curate-text/process-data/quality-assessment/distributed-classifier.html) documentation page."
"Before running this notebook, see this [Installation Guide](https://docs.nvidia.com/nemo/curator/latest/admin/installation.html#admin-installation) page for instructions on how to install NeMo Curator. Be sure to use an installation method which includes GPU dependencies (`text_cuda12` or `all`). Check proper installation with:"
]
},
{
"cell_type": "code",
"execution_count": 1,
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"# Silence Curator logs via Loguru\n",
"# First, silence Curator logs via Loguru\n",
"import os\n",
"\n",
"os.environ[\"LOGURU_LEVEL\"] = \"ERROR\""
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"import nemo_curator\n",
"\n",
"nemo_curator.__version__ # should be >= 1.0.0"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"We can check that GPUs are available, then check that the `gpustat` dependency was installed:"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"!nvidia-smi"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"import gpustat\n",
"\n",
"gpustat.__version__ # check gpu dependency is installed"
]
},
{
"cell_type": "markdown",
"metadata": {},
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -12,23 +12,61 @@
" - Volta™ or higher (compute capability 7.0+)\n",
" - CUDA 12.x\n",
"\n",
"Before running this notebook, see this [Installation Guide](https://docs.nvidia.com/nemo/curator/latest/admin/installation.html#admin-installation) page for instructions on how to install NeMo Curator. Be sure to use an installation method which includes GPU dependencies.\n",
"For more information about the classifiers, refer to our [Distributed Data Classification](https://docs.nvidia.com/nemo/curator/latest/curate-text/process-data/quality-assessment/distributed-classifier.html) documentation page.\n",
"\n",
"For more information about the classifiers, refer to our [Distributed Data Classification](https://docs.nvidia.com/nemo/curator/latest/curate-text/process-data/quality-assessment/distributed-classifier.html) documentation page."
"Before running this notebook, see this [Installation Guide](https://docs.nvidia.com/nemo/curator/latest/admin/installation.html#admin-installation) page for instructions on how to install NeMo Curator. Be sure to use an installation method which includes GPU dependencies (`text_cuda12` or `all`). Check proper installation with:"
]
},
{
"cell_type": "code",
"execution_count": 1,
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"# Silence Curator logs via Loguru\n",
"# First, silence Curator logs via Loguru\n",
"import os\n",
"\n",
"os.environ[\"LOGURU_LEVEL\"] = \"ERROR\""
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"import nemo_curator\n",
"\n",
"nemo_curator.__version__ # should be >= 1.0.0"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"We can check that GPUs are available, then check that the `gpustat` dependency was installed:"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"!nvidia-smi"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"import gpustat\n",
"\n",
"gpustat.__version__ # check gpu dependency is installed"
]
},
{
"cell_type": "markdown",
"metadata": {},
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -12,23 +12,61 @@
" - Volta™ or higher (compute capability 7.0+)\n",
" - CUDA 12.x\n",
"\n",
"Before running this notebook, see this [Installation Guide](https://docs.nvidia.com/nemo/curator/latest/admin/installation.html#admin-installation) page for instructions on how to install NeMo Curator. Be sure to use an installation method which includes GPU dependencies.\n",
"For more information about the classifiers, refer to our [Distributed Data Classification](https://docs.nvidia.com/nemo/curator/latest/curate-text/process-data/quality-assessment/distributed-classifier.html) documentation page.\n",
"\n",
"For more information about the classifiers, refer to our [Distributed Data Classification](https://docs.nvidia.com/nemo/curator/latest/curate-text/process-data/quality-assessment/distributed-classifier.html) documentation page."
"Before running this notebook, see this [Installation Guide](https://docs.nvidia.com/nemo/curator/latest/admin/installation.html#admin-installation) page for instructions on how to install NeMo Curator. Be sure to use an installation method which includes GPU dependencies (`text_cuda12` or `all`). Check proper installation with:"
]
},
{
"cell_type": "code",
"execution_count": 1,
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"# Silence Curator logs via Loguru\n",
"# First, silence Curator logs via Loguru\n",
"import os\n",
"\n",
"os.environ[\"LOGURU_LEVEL\"] = \"ERROR\""
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"import nemo_curator\n",
"\n",
"nemo_curator.__version__ # should be >= 1.0.0"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"We can check that GPUs are available, then check that the `gpustat` dependency was installed:"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"!nvidia-smi"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"import gpustat\n",
"\n",
"gpustat.__version__ # check gpu dependency is installed"
]
},
{
"cell_type": "markdown",
"metadata": {},
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -12,23 +12,61 @@
" - Volta™ or higher (compute capability 7.0+)\n",
" - CUDA 12.x\n",
"\n",
"Before running this notebook, see this [Installation Guide](https://docs.nvidia.com/nemo/curator/latest/admin/installation.html#admin-installation) page for instructions on how to install NeMo Curator. Be sure to use an installation method which includes GPU dependencies.\n",
"For more information about the classifiers, refer to our [Distributed Data Classification](https://docs.nvidia.com/nemo/curator/latest/curate-text/process-data/quality-assessment/distributed-classifier.html) documentation page.\n",
"\n",
"For more information about the classifiers, refer to our [Distributed Data Classification](https://docs.nvidia.com/nemo/curator/latest/curate-text/process-data/quality-assessment/distributed-classifier.html) documentation page."
"Before running this notebook, see this [Installation Guide](https://docs.nvidia.com/nemo/curator/latest/admin/installation.html#admin-installation) page for instructions on how to install NeMo Curator. Be sure to use an installation method which includes GPU dependencies (`text_cuda12` or `all`). Check proper installation with:"
]
},
{
"cell_type": "code",
"execution_count": 1,
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"# Silence Curator logs via Loguru\n",
"# First, silence Curator logs via Loguru\n",
"import os\n",
"\n",
"os.environ[\"LOGURU_LEVEL\"] = \"ERROR\""
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"import nemo_curator\n",
"\n",
"nemo_curator.__version__ # should be >= 1.0.0"
]
},
{
"cell_type": "markdown",
"metadata": {},
"source": [
"We can check that GPUs are available, then check that the `gpustat` dependency was installed:"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"!nvidia-smi"
]
},
{
"cell_type": "code",
"execution_count": null,
"metadata": {},
"outputs": [],
"source": [
"import gpustat\n",
"\n",
"gpustat.__version__ # check gpu dependency is installed"
]
},
{
"cell_type": "markdown",
"metadata": {},
Expand Down
Loading
Loading