Code

New design of the transformer API #1022

Code

Merged

karlhigley merged 14 commits intomainNVIDIA-Merlin/models:mainfrom transformer-apiNVIDIA-Merlin/models:transformer-apiCopy head branch name to clipboardMar 21, 2023

+876-179

Commits

Commits on Mar 16, 2023

• implement new design of the Transformer API on top of the release-23.02 branch

  sararb

  committed
  b6f1809

  Copy full SHA for b6f1809

• add support of ragged tensor to weight tying

  sararb

  committed
  472f2b7

  Copy full SHA for 472f2b7

• update example notebook with the new API

  sararb

  committed
  6ad7a3d

  Copy full SHA for 6ad7a3d

• include PR comments

  sararb

  committed
  45d0401

  Copy full SHA for 45d0401

• fix masking of sequence-predict-next transform

  sararb

  committed
  f122bb3

  Copy full SHA for f122bb3

• adjust sample_weights to targets shape

  sararb

  committed
  afcb735

  Copy full SHA for afcb735

• add masking support to SequencePredictRandom transform

  sararb

  committed
  8dd2384

  Copy full SHA for 8dd2384

• rebase with main branch to include data loader changes

  sararb

  committed
  b9af543

  Copy full SHA for b9af543

• fix linting

  sararb

  committed
  ac5efe1

  Copy full SHA for ac5efe1

Commits on Mar 17, 2023

• Fix the adjust-predictions logic to support targets as 2-D scalars

  sararb

  committed
  61fd1da

  Copy full SHA for 61fd1da

• Fix transformer example notebook

  sararb

  committed
  df5a004

  Copy full SHA for df5a004

Commits on Mar 20, 2023

• Merge branch 'main' into transformer-api

  sararb

  authored
  af2d519

  Copy full SHA for af2d519

• update import of transformer blocks in transforms/sequence and move them inside configure_for_train() function.

  sararb

  committed
  349a5f9

  Copy full SHA for 349a5f9

Commits on Mar 21, 2023

• Merge branch 'main' into transformer-api

  sararb

  authored
  a2fdb68

  Copy full SHA for a2fdb68