hypersound/requirements.txt at main · WUT-AI/hypersound · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
# Generic
jupyterlab==3.2.*
jupyterlab-lsp==3.9.*
librosa==0.9.2
matplotlib
more-itertools
numpy==1.22.*
packaging
pandas==1.3.*
pathos==0.2.*
pyparsing==2.4.7
rich
scikit-learn
scipy==1.8.*
seaborn
soundfile
typer==0.4.*
wandb==0.12.*

# Code quality
black
black[jupyter]
flake8
flake8-annotations-complexity
flake8-cognitive-complexity
flake8-docstrings
flake8-simplify
isort[colors]
mypy
pydocstyle
pylint
pytest

# PyTorch & Hydra
--find-links https://download.pytorch.org/whl/cu113/torch_stable.html
torch==1.11.*
torchaudio==0.11.*
torchmetrics==0.9.3
torchvision==0.12.*
hydra-core==1.2.*
git+https://github.com/pytorch/hydra-torch/#subdirectory=hydra-configs-torch
git+https://github.com/pytorch/hydra-torch/#subdirectory=hydra-configs-torchvision
pytorch-lightning==1.5.*
lightning-bolts==0.4.*
pytorch-yard==2022.9.1
torch_tb_profiler==0.3.*
torchinfo==1.5.*
hypnettorch==0.0.4
auraloss==0.2.*
cdpam==0.0.6
pystoi==0.3.3
pesq==0.0.4

# RAVE
effortless-config==0.7.0
einops==0.4.0
GPUtil==1.4.0
tqdm==4.62.3
git+https://github.com/caillonantoine/cached_conv.git@v2.3.6#egg=cached_conv
git+https://github.com/caillonantoine/UDLS.git@v1.0.0#egg=udls