Skip to content

Commit

Permalink
rename
Browse files Browse the repository at this point in the history
  • Loading branch information
priyakasimbeg committed Nov 21, 2023
1 parent 30a0654 commit b24e7e7
Show file tree
Hide file tree
Showing 10 changed files with 2,961 additions and 0 deletions.
83 changes: 83 additions & 0 deletions reference_algorithms/prize_qualification_baselines/README.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,83 @@
# Prize Qualification Baselines
This directory contains the baseine(s) that submissions that must beat to qualify for prizes.

TODO: link back to section in rules.

## Externally Tuned Ruleset

### JAX

The prize qualification baseline submissions for jax are:
- `reference_algorithms/prize_qualification_baselines/external_tuning/jax_nadamw_target_setting.py`
- `feference_algorithms/prize_qualification_baselines/external_tuning/jax_nadamw_full_budget.py`

Example command:

```bash
python3 submission_runner.py \
--framework=jax \
--data_dir=<data_dir> \
--experiment_dir=<experiment_dir> \
--experiment_name=<experiment_name> \
--workload=<workload> \
--submission_path=reference_algorithms/prize_qualification_baselines/external_tuning/jax_nadamw_target_setting.py \
--tuning_search_space=reference_algorithms/prize_qualification_baselines/external_tuning/tuning_search_space.json
```

### PyTorch

The prize qualification baseline submissionss for PyTorch are:
- `reference_algorithms/prize_qualification_baselines/external_tuning/pytorch_nadamw_target_setting.py`
- `feference_algorithms/prize_qualification_baselines/external_tuning/pytorch_nadamw_full_budget.py`


Example command:

```bash
torchrun --redirects 1:0,2:0,3:0,4:0,5:0,6:0,7:0 --standalone --nnodes=1 --nproc_per_node=8 submission_runner.py \
--framework=pytorch \
--data_dir=<data_dir> \
--experiment_dir=<experiment_dir> \
--experiment_name=t<experiment_name> \
--workload=<workload>\
--submission_path=reference_algorithms/prize_qualification_baselines/external_tuning/pytorch_nadamw_target_setting.py \
--tuning_search_space=reference_algorithms/prize_qualification_baselines/external_tuning/tuning_search_space.json
```

## Self-tuning Ruleset

### JAX

The prize qualification baseline submissionss for jax are:
- `reference_algorithms/prize_qualification_baselines/external_tuning/jax_nadamw_target_setting.py`
- `feference_algorithms/prize_qualification_baselines/external_tuning/jax_nadamw_full_budget.py`

Example command:
```bash
python3 submission_runner.py \
--framework=jax \
--data_dir=<data_dir> \
--experiment_dir=<experiment_dir> \
--experiment_name=<experiment_name> \
--workload=<workload> \
--submission_path=reference_algorithms/prize_qualification_baselines/external_tuning/jax_nadamw_target_setting.py \
--tuning_ruleset=self
```

### PyTorch

The prize qualification baseline submissionss for PyTorch are:
- `reference_algorithms/prize_qualification_baselines/external_tuning/pytorch_nadamw_target_setting.py`
- `feference_algorithms/prize_qualification_baselines/external_tuning/pytorch_nadamw_full_budget.py`

Example command:
```bash
torchrun --redirects 1:0,2:0,3:0,4:0,5:0,6:0,7:0 --standalone --nnodes=1 --nproc_per_node=8 submission_runner.py \
--framework=pytorch \
--data_dir=<data_dir> \
--experiment_dir=<experiment_dir> \
--experiment_name=t<experiment_name> \
--workload=<workload>\
--submission_path=reference_algorithms/prize_qualification_baselines/external_tuning/pytorch_nadamw_target_setting.py \
--tuning_ruleset=self
```
Loading

0 comments on commit b24e7e7

Please sign in to comment.