Aman Gupta

icon indicating an email [email protected]

icon company LinkedIn icon location United States icon location Sr Staff ML guy at LinkedIn. Enthusiastic about ML, Optimization and Large-scale systems

Results 2 issues of Aman Gupta

Modify generate.sh to take a dynamic number of GPUs as input

1 comment

## Issue Data generation requires exactly 8 GPUs to be present. This doesn't make the code run properly on machines with less than 8 GPUs (for instance, I am using...

Added a new PR to allow generation on fewer than 8 GPUs

Hello authors, Great work! I added a quick PR to adapt generation to run on fewer than 8 GPUs if needed - https://github.com/uclaml/SPPO/pull/24. This is a minimally invasive change