Code

Add baselines to ViT tests#365

Code

Open

ashors1 wants to merge 14 commits intomainNVIDIA/JAX-Toolbox:mainfrom vit-add-baselinesNVIDIA/JAX-Toolbox:vit-add-baselinesCopy head branch name to clipboard

+78-7

Commits

Commits on Nov 1, 2023

• [WIP] add loss and perf baselines to vit tests

  ashors1

  committed
  ec458f5

  Copy full SHA for ec458f5

Commits on Nov 6, 2023

• add vit baselines and update metrics tests to support vit

  ashors1

  committed
  0a6b0cf

  Copy full SHA for 0a6b0cf

Commits on Nov 7, 2023

• fix typo

  ashors1

  committed
  d88362a

  Copy full SHA for d88362a

• fix typo

  ashors1

  committed
  2078bff

  Copy full SHA for 2078bff

• fix e2e_time_mult key

  ashors1

  committed
  9e722fd

  Copy full SHA for 9e722fd

• Merge branch 'main' of github.com:NVIDIA/JAX-Toolbox into vit-add-baselines

  ashors1

  committed
  bb22e01

  Copy full SHA for bb22e01

• update e2e tolerance for vit 8G1N test

  ashors1

  committed
  241ba80

  Copy full SHA for 241ba80

Commits on Nov 8, 2023

• Merge branch 'main' of github.com:NVIDIA/JAX-Toolbox into vit-add-baselines

  ashors1

  committed
  34af83e

  Copy full SHA for 34af83e

Commits on Nov 27, 2023

• Merge branch 'main' of github.com:NVIDIA/JAX-Toolbox into vit-add-baselines

  ashors1

  committed
  81996e5

  Copy full SHA for 81996e5

Commits on Nov 30, 2023

• Merge branch 'main' of github.com:NVIDIA/JAX-Toolbox into vit-add-baselines

  ashors1

  committed
  bd7ba7c

  Copy full SHA for bd7ba7c

Commits on Dec 4, 2023

• Merge branch 'main' of github.com:NVIDIA/JAX-Toolbox into vit-add-baselines

  ashors1

  committed
  4e08ca9

  Copy full SHA for 4e08ca9

Commits on Dec 5, 2023

• Merge branch 'main' of github.com:NVIDIA/JAX-Toolbox into vit-add-baselines

  ashors1

  committed
  4bf6837

  Copy full SHA for 4bf6837

Commits on Dec 18, 2023

• Merge branch 'main' of github.com:NVIDIA/JAX-Toolbox into vit-add-baselines

  ashors1

  committed
  0355183

  Copy full SHA for 0355183

• fix typo

  ashors1

  committed
  542b9ab

  Copy full SHA for 542b9ab