Code

Fix shard_and_maybe_pad_np function#515

Code

Merged

priyakasimbeg merged 3 commits intomlcommons:devmlcommons/algorithmic-efficiency:devfrom runame:fix-paddingruname/algorithmic-efficiency:fix-paddingCopy head branch name to clipboardSep 21, 2023

+15-16

Commits

Commits on Sep 13, 2023

• Simplify pad function

  runame

  committed
  2f76cb9

Commits on Sep 14, 2023

• Always pad to global_batch_size when it is provided

  runame

  committed
  35c8736

• Fix pad_size in pad function

  runame

  committed
  ad64fd1