diff options
| author | Marvin | 2023-08-18 12:59:17 +0200 |
|---|---|---|
| committer | GitHub | 2023-08-18 12:59:17 +0200 |
| commit | ff574af3cfb4ebfc2c1faf8cdc4bf87560dd8373 (patch) | |
| tree | cb4f912bc3a9a2c5e8ba2b8567fe2c8c785cf661 /pyproject.toml | |
| parent | 58f07302989f945b109c0c69d9e96444514ef1a6 (diff) | |
| parent | 2bcaf9c687dfc3a19d0409361a3dc9b535150e8a (diff) | |
Merge pull request #7 from Algo-Boys/feature-dataloading
Feature dataloading
Diffstat (limited to 'pyproject.toml')
| -rw-r--r-- | pyproject.toml | 26 |
1 files changed, 14 insertions, 12 deletions
diff --git a/pyproject.toml b/pyproject.toml index 928874a..9c09a99 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -11,13 +11,10 @@ packages = [{include = "swr2_asr"}] python = "^3.10" numpy = "^1.25.2" click = "^8.1.6" - -[tool.poetry.group.cpu] -optional = true - -[tool.poetry.group.cpu.dependencies] -torch = "=2.0.1" -torchaudio = "=2.0.2" +audioloader = {git = "https://github.com/KinWaiCheuk/AudioLoader.git"} +tqdm = "^4.66.1" +torch = {version = "^2.0.1+cpu", source = "pytorch-cpu"} +torchaudio = {version = "^2.0.2+cpu", source = "pytorch-cpu"} [tool.poetry.group.dev.dependencies] black = "^23.7.0" @@ -26,18 +23,23 @@ mypy = "^1.5.1" pre-commit = "^3.3.3" [[tool.poetry.source]] -name = "pytorch" -url = "https://download.pytorch.org/whl/cu111/" -priority = "supplemental" +name = "pytorch-cpu" +url = "https://download.pytorch.org/whl/cpu" +priority = "explicit" + [[tool.poetry.source]] -name = "PyPI" -priority = "primary" +name = "pytorch-gpu" +url = "https://download.pytorch.org/whl/cu118" +priority = "explicit" [[tool.mypy.overrides]] module = "torchaudio.*" ignore_missing_imports = true +[tool.poetry.scripts] +train = "swr2_asr.train:main" + [build-system] requires = ["poetry-core"] build-backend = "poetry.core.masonry.api" |