tdpeuter/2025ML-project-neural_compression
tdpeuter/2025ML-project-neural_compression
Archived
1
Fork 0
Code Issues Pull requests Releases 1 Activity Actions

Add testing configs

Browse source
This commit is contained in:
Tibo De Peuter 2025-12-11 22:21:47 +01:00
parent 5de8181959
commit ff11c1deb3
Signed by: tdpeuter
GPG key ID: 38297DE43F75FFE2
9 changed files with 250 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

17
README.md
View file

@ -3,9 +3,22 @@
Example usage:
```shell
python main.py --debug train --dataset enwik9 --data-root ~/data/datasets/ml --method optuna --model transformer --model-save-path ~/data/ml-models/test-transformer.pt
# Fetching
python main.py --debug train --method fetch \
--dataset enwik9 --data-root /path/to/datasets
python benchmark.py --debug train --dataset enwik9 --data-root ~/data/datasets/ml --method optuna --model cnn --model-save-path ~/data/ml-models/test-cnn.pt
# Training
python main.py --debug train --method optuna \
--dataset enwik9 --data-root /path/to/datasets \
--model cnn --model-save-path /path/to/optuna-model
python main.py --debug --results /path/to/results train --method full \
--dataset enwik9 --data-root /path/to/datasets \
--model-load-path /path/to/optuna-model --model-save-path /path/to/full-model
# Compressing
python benchmark.py --debug compress \
--model-load-path /path/to/full-model \
--input-file inputfile --output-file outputfile
```
## Running locally