updated README

This commit is contained in:
2026-01-10 20:49:59 +01:00
parent d6df888387
commit 7a04a3ac67

View File

@@ -1,3 +1,31 @@
# caveman # caveman
Audio reconstruction model, that "uncompresses" lossy mp3s. Audio reconstruction model, that "uncompresses" lossy mp3s.
## Installing dependencies
There is the pyproject file, so you can use that.
Or, if you prefer to keep your sanity, use uv:
```
uv sync
```
## Training
Adjust the location of your data directories via the variables in the cavemanml.py file. Any compression type is accepted, as long as your audio is 44100kHz.
```
uv run cavemanml.py
```
Obviously the original dataset (FMA) is not provided with this repo.
## Inference
For now, the only inference this code can do, is the example. You can adjust the input and output file in the code, though. Theoretically, the model should take on anything. To use the provided checkpoint:
```
uv run cavemanml.py ./checkpoint70.pth
```