Skip to content

Josh1108/EPtest

Repository files navigation

Installation

Run the following commands:

git clone https://github.com/Josh1108/EPtest/tree/main
pip install -r requirements.txt

# Add the following to your .bash_rc or .bash_profile 
export PYTHONPATH=/path/to/eptest/root/folder:$PYTHONPATH

Running experiments

.
├── converters/              # files to convert data in EPtest fromat
├── core/                    # all core functionailites
│   ├── trainer.py           # training file
│   ├── test.py              # test file
│   ├── models/              # models used for EP tests
|   └── config/              # configs used for training and testing
└── ...

To train models, run:

python3 trainer.py --config_file <PATH_OF_FILE>

For testing, run:

python3 test.py --config_file <PATH_OF_FILE>

Datasets

We are still figuring out copyright issues with the datasets, in the meantime, if you want access, please email us at sagnikrayc at gmail.

About

Edge Probing tests for LLM's

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published