Skip to content

Commit ca17609

Browse files
Fix benchmark_low_bit_adam.py reference (#2287)
Signed-off-by: Emmanuel Ferdman <emmanuelferdman@gmail.com>
1 parent ce27731 commit ca17609

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torchao/optim/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ NOTE:
3232

3333
## Benchmarks
3434

35-
Fine-tune [timm](https://github.com/huggingface/pytorch-image-models)'s [ViT-H](https://huggingface.co/timm/vit_huge_patch14_224.orig_in21k) (630M params) on [resisc45](https://huggingface.co/datasets/timm/resisc45) dataset. PyTorch 2.4, BF16 AMP, compiled model, 1 epoch, batch size 8, cosine LR scheduler, 4070Ti SUPER, fixed random seed. Benchmark script is available at [benchmarks/benchmark_low_bit_adam.py](../../../benchmarks/benchmark_low_bit_adam.py).
35+
Fine-tune [timm](https://github.com/huggingface/pytorch-image-models)'s [ViT-H](https://huggingface.co/timm/vit_huge_patch14_224.orig_in21k) (630M params) on [resisc45](https://huggingface.co/datasets/timm/resisc45) dataset. PyTorch 2.4, BF16 AMP, compiled model, 1 epoch, batch size 8, cosine LR scheduler, 4070Ti SUPER, fixed random seed. Benchmark script is available at [benchmarks/benchmark_low_bit_adam.py](../../benchmarks/benchmark_low_bit_adam.py).
3636

3737
AdamW impl | Peak memory allocated (GB) | imgs/s | accuracy
3838
----------------|----------------------------|--------|----------

0 commit comments

Comments
 (0)