Update and rename training.md to models/README.md
This commit is contained in:
parent
834c0bd238
commit
8ab1741e4c
1 changed files with 11 additions and 1 deletions
19
models/README.md
Normal file
19
models/README.md
Normal file
|
|
@ -0,0 +1,19 @@
|
|||
# Meta information about training
|
||||
|
||||
The trained models that are saved here follow the following naming convention:
|
||||
|
||||
* Optuna intermediate model: `[model]-[dataset]-[context].pt`
|
||||
* Fully trained model: `[model]-[dataset]-full-[context].pt`
|
||||
|
||||
The following parameters were used:
|
||||
|
||||
* training size: 2048 for optuna, 209715 for full training
|
||||
* context sizes: {128, 256}
|
||||
|
||||
The models were trained with the following command:
|
||||
|
||||
```bash
|
||||
uv run python ./results/[cnn,autoencoder] train --method [full,optuna] \
|
||||
--data-root ./data --dataset [genome,enwik9] --context [128,256] --size [2048,209715] \
|
||||
--model-save-path ./models/<name> --model-load-path <path if full trainig, output from optuna>
|
||||
```
|
||||
Reference in a new issue