Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Initial Refactor and add new features for SAT solver including gpt_train.py and downstream tasks, plus addition of multi-node/multi-GPU processes #2

Open
wants to merge 3 commits into
base: main
Choose a base branch
from

Conversation

cesposo
Copy link
Collaborator

@cesposo cesposo commented Oct 2, 2024

This PR refactors key portions of the code, improves performance, and adds new features for SAT problem solving. Specific changes include:

  • Extended/refactored gpt_train.py to include multiple model sizes, mostly for experiments with scale-laws
  • Changed ancillary scripts downstream to gpt_train.py changes which includes alterations to tokenization previously made
  • Added support for batch processing in folder_eval.py for Windows and Linux (eval_folder.py and eval_folder.sh)
  • Improved logging and error handling during model evaluation.
  • Addition of multi-node/multi-GPU batch processing for supercomputer cluster (experimental)

These changes aim to improve code maintainability and add new functionality for more flexible SAT problem solving.

leyanpan and others added 3 commits September 28, 2024 19:47
…f model sizes for our experiments with respect to to scaling laws. These also include hte updated tokenization/encoding schemes. Likewise, SAT_DATASET and Configurator were also modified to accomodate the new tokenization/encoding scheme. Further, we added a folder_eval.py which can be run in lieu of the bash script on windows machines for validation for out-of-distribution tests.
…as a basic example of running the code on multi-node/multi-gpu for larger LLMs
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants