Update Criteo1TB eval bsz for pytorch and jax

mlcommons / algorithmic-efficiency

MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvements in both training algorithms and models.

https://mlcommons.org/en/groups/research-algorithms/

Apache License 2.0

321 stars 62 forks source link

Update Criteo1TB eval bsz for pytorch and jax #690

Closed priyakasimbeg closed 6 months ago

priyakasimbeg commented 6 months ago

Addresses https://github.com/mlcommons/algorithmic-efficiency/issues/641

Update Criteo1TB eval bsz for PyTorch and JAX to reduce likelihood of OOMs during eval. Note this change won't affect submission times measured for scoring since the submission time excludes evals.

github-actions[bot] commented 6 months ago

MLCommons CLA bot All contributors have signed the MLCommons CLA ✍️ ✅