Code

Contrastive Reranker/Reward model #9171

Code

Merged

arendu merged 26 commits intomainNVIDIA/NeMo:mainfrom adithyare/embedding_rerankingNVIDIA/NeMo:adithyare/embedding_rerankingCopy head branch name to clipboardJul 10, 2024

+1,115-51

Commits

Commits on May 8, 2024

• wip contrastive reranker

  Show description for ff88d62
  arendu

  committed
  ff88d62

Commits on May 9, 2024

• wip

  Show description for 6a2b005
  arendu

  committed
  6a2b005

• wip

  Show description for f818c3c
  arendu

  committed
  f818c3c

Commits on May 11, 2024

• working reranker training and validation

  Show description for 52203b9
  arendu

  committed
  52203b9

• default peft for reranker

  Show description for 4ac7642
  arendu

  committed
  4ac7642

• validation time update

  Show description for b67d0e7
  arendu

  committed
  b67d0e7

• reranker test

  Show description for 7ecddf6
  arendu

  committed
  7ecddf6

Commits on May 12, 2024

• reranker inference

  Show description for cebd9d8
  arendu

  committed
  cebd9d8

• reranker inference

  Show description for d1c44a8
  arendu

  committed
  d1c44a8

• Merge branch 'main' into adithyare/embedding_reranking

  arendu

  authored
  6b3a56f

• Apply isort and black reformatting

  Show description for 13a6f31
  arendu

  committed
  13a6f31

Commits on May 13, 2024

• updates

  Show description for 05b4d17
  arendu

  committed
  05b4d17

• Apply isort and black reformatting

  Show description for 2bf75c7
  arendu

  committed
  2bf75c7

Commits on May 14, 2024

• updates

  Show description for 4c81ec2
  arendu

  committed
  4c81ec2

• Apply isort and black reformatting

  Show description for af79ef8
  arendu

  committed
  af79ef8

• Merge branch 'main' into adithyare/embedding_reranking

  arendu

  authored
  c591f9f

Commits on May 15, 2024

• also can support rlhf style reward model loss

  Show description for a3032b6
  arendu

  committed
  a3032b6

• Merge branch 'main' into adithyare/embedding_reranking

  arendu

  authored
  cc83dc1

Commits on May 20, 2024

• Merge branch 'main' into adithyare/embedding_reranking

  arendu

  authored
  d975416

Commits on May 31, 2024

• Merge branch 'main' into adithyare/embedding_reranking

  Show description for 19ccebf
  arendu

  authored
  19ccebf

• Apply isort and black reformatting

  Show description for c1e43d8
  arendu

  committed
  c1e43d8

Commits on Jul 9, 2024

• Merge branch 'main' into adithyare/embedding_reranking

  Show description for badebe8
  arendu

  authored
  badebe8

• Apply isort and black reformatting

  Show description for 173b93b
  arendu

  committed
  173b93b

• Merge branch 'main' into adithyare/embedding_reranking

  arendu

  authored
  97b8dfd

Commits on Jul 10, 2024

• typo in cicd

  Show description for db3934d
  arendu

  committed
  db3934d

• Merge branch 'main' into adithyare/embedding_reranking

  arendu

  authored
  3df4c64