pytorch
e5e451a9 - Update batch size for a couple models (#101837)

Commit
1 year ago
Update batch size for a couple models (#101837) The memory compression for these models is at parity, but because we interleave timings between torch.compile and eager run memory is duplicated between between eager and cudagraphs pool and causes OOM. Pull Request resolved: https://github.com/pytorch/pytorch/pull/101837 Approved by: https://github.com/anijain2305
Author
Committer
Parents
Loading