Code

Add BERT Embedding Models#11737

Code

Merged

suiyoubi merged 18 commits intomainNVIDIA/NeMo:mainfrom aot/bert-embedding-modelsNVIDIA/NeMo:aot/bert-embedding-modelsCopy head branch name to clipboardJan 6, 2025

+794-5

Commits

Commits on Dec 10, 2024

• Add BERT Embedding model

  suiyoubi

  committed
  514af14

  Copy full SHA for 514af14

Commits on Dec 11, 2024

• Fix validation loss func

  suiyoubi

  committed
  7458e11

  Copy full SHA for 7458e11

Commits on Jan 2, 2025

• Revert lightning/io changes

  suiyoubi

  committed
  b564066

  Copy full SHA for b564066

• Revert llm/api

  suiyoubi

  committed
  dfca125

  Copy full SHA for dfca125

Commits on Jan 3, 2025

• add global in-batch negatives option

  suiyoubi

  committed
  7ca572a

  Copy full SHA for 7ca572a

• Apply isort and black reformatting

  Show description for 91e6ac8
  suiyoubi

  committed
  91e6ac8

  Copy full SHA for 91e6ac8

• codeQL

  suiyoubi

  committed
  0a54c4d

  Copy full SHA for 0a54c4d

• pylint

  suiyoubi

  committed
  6df7f5d

  Copy full SHA for 6df7f5d

• Merge remote-tracking branch 'origin/main' into aot/bert-embedding-models

  suiyoubi

  committed
  ad24c17

  Copy full SHA for ad24c17

• Apply isort and black reformatting

  Show description for 60837c1
  suiyoubi

  committed
  60837c1

  Copy full SHA for 60837c1

• Merge remote-tracking branch 'origin/aot/bert-embedding-models' into aot/bert-embedding-models

  suiyoubi

  committed
  2c6f63a

  Copy full SHA for 2c6f63a

• pylint

  suiyoubi

  committed
  5122ded

  Copy full SHA for 5122ded

• minor changes

  suiyoubi

  committed
  8fe6533

  Copy full SHA for 8fe6533

• Apply isort and black reformatting

  Show description for 64e1289
  suiyoubi

  committed
  64e1289

  Copy full SHA for 64e1289

• remove test code for specter

  suiyoubi

  committed
  b1b4aa8

  Copy full SHA for b1b4aa8

• remove import

  suiyoubi

  committed
  e9da3a8

  Copy full SHA for e9da3a8

Commits on Jan 5, 2025

• Copy EmbeddingHead to nemo2

  suiyoubi

  committed
  e497e5b

  Copy full SHA for e497e5b

• Apply isort and black reformatting

  Show description for d5ba8f3
  suiyoubi

  committed
  d5ba8f3

  Copy full SHA for d5ba8f3