Skip to main content
ContainersNVIDIADeepSeek-R1NVIDIALlama-3.1-Nemotron-70B-InstructNVIDIAPyTorch
CollectionsNVIDIAOmniverse Kit (FB)NVIDIADeepStream SDKNVIDIAOmniverse Kit App Streaming
ModelsNVIDIAStyleGAN3 pretrained modelsNVIDIAPeopleNetNVIDIATrafficCamNet
ResourcesNVIDIARiva Skills Quick Start
Helm ChartsNVIDIAGPU OperatorNVIDIANVIDIA NIM Operator
  • NVIDIA
    Text Classification with BERT and NeMo
    Container
    NVIDIA
    Text Classification with BERT and NeMo

    Text Classification with BERT and NeMo. This NeMo application trains text classification models using single-GPU or multi-GPU. We log performance metrics and visualize them with TensorBoard. We show how to do inference with NeMo, and we visualize BERT embeddings before and after fine-tuning.

  • LayerLabelCreated
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4CMD
    ["-c" "jupyter lab  --notebook-dir=/NeMo-0.10.1/examples/nlp/text_classification --ip=0.0.0.0 --no-browser --allow-root --port=8888 --NotebookApp.token='' --NotebookApp.password='' --NotebookApp.allow_origin='*' --NotebookApp.base_url=${NB_PREFIX}"]
    		07/09/2020 9:16 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENTRYPOINT
    /bin/sh
    		07/09/2020 9:16 PM UTC
    sha256:549f6b2c1081ae734310ec90dc046892e2b98f9b032ed263ba3fd14832bde3a6COPY
    dir:d65c1de4f102bb86821db949911035b12d54f6071388fc716744d5a4642ff3bf in figures
    		07/09/2020 9:16 PM UTC
    sha256:7c1dd197f6d36d227b8864b9651368dcb4e6658a0b22be9b53e04e17ffa54d04RUN
    mkdir figures
    		07/09/2020 9:16 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4WORKDIR
    /NeMo-0.10.1/examples/nlp/text_classification
    		07/09/2020 9:16 PM UTC
    sha256:d17d7465023891dc549571b8b13e4352533c4094f082ff3876b5eebd1eff3a43COPY
    file:18974cc708050d04d8cdac497c31e58e30b5338b979c10dab3db6cdbc5e1cb87 in /NeMo-0.10.1/examples/nlp/text_classification
    		07/09/2020 9:16 PM UTC
    sha256:24bf80a13d011583fd757db11e760136c5465136763138595ee08da8c42d193cCOPY
    dir:a480cbec96ba5ec0d812a778d9f3c79b39914c17ab05a554257dd500ff7aebf3 in /root/.jupyter
    		07/09/2020 9:16 PM UTC
    sha256:3667b33d1512843b6a9ece0aa14e395bee6696fbbd97302ad249979547987712RUN
    cd /NeMo-0.10.1 &&
  bash reinstall.sh
    		07/09/2020 9:16 PM UTC
    sha256:e00ec5958149ac9f3ff53e266e215fbcce4dcb65bdea4d0da1692fcba302c054COPY
    dir:d5c7c627e572c85a1d208484c1364669bf468865cf2137959f9910d61b3d462b in /NeMo-0.10.1
    		07/09/2020 9:15 PM UTC
    sha256:668ce0158e822dbc2d1e1436cfe3f79d2ae5531d7ef4c24595a834806270a33bRUN
    mkdir /NeMo-0.10.1
    		07/09/2020 9:15 PM UTC
    ...

    NVIDIA uses cookies to improve your experience on our web site. We and our third-party partners also use cookies and other tools to collect and record information you provide as well as information about your interactions with our websites for performance improvement, analytics, and to assist in marketing efforts. By clicking "Accept All", you consent to our use of cookies and other tools as described in our Cookie Policy. You can manage your cookie settings by clicking on "Manage Settings." By continuing to use this site or by clicking one of the buttons below, you agree to our Terms of Service (which contains important waivers). Please see our Privacy Policy for more information on our privacy practices.