summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--poetry.lock25
-rw-r--r--pyproject.toml2
2 files changed, 16 insertions, 11 deletions
diff --git a/poetry.lock b/poetry.lock
index 9102d0d..c2e7975 100644
--- a/poetry.lock
+++ b/poetry.lock
@@ -482,7 +482,7 @@ flake8 = "*"
[[package]]
name = "fsspec"
-version = "2021.8.1"
+version = "2021.9.0"
description = "File-system specification"
category = "main"
optional = false
@@ -495,15 +495,20 @@ requests = {version = "*", optional = true, markers = "extra == \"http\""}
[package.extras]
abfs = ["adlfs"]
adl = ["adlfs"]
+arrow = ["pyarrow (>=1)"]
dask = ["dask", "distributed"]
dropbox = ["dropboxdrivefs", "requests", "dropbox"]
entrypoints = ["importlib-metadata"]
+fuse = ["fusepy"]
gcs = ["gcsfs"]
git = ["pygit2"]
github = ["requests"]
gs = ["gcsfs"]
+gui = ["panel"]
hdfs = ["pyarrow (>=1)"]
http = ["requests", "aiohttp"]
+libarchive = ["libarchive-c"]
+oci = ["ocifs"]
s3 = ["s3fs"]
sftp = ["paramiko"]
smb = ["smbprotocol"]
@@ -1896,10 +1901,10 @@ python-versions = ">=3.6.2"
typing-extensions = "*"
[[package]]
-name = "torch-summary"
-version = "1.4.5"
+name = "torchinfo"
+version = "1.5.3"
description = "Model summary in PyTorch, based off of the original torchsummary."
-category = "main"
+category = "dev"
optional = false
python-versions = ">=3.6"
@@ -2133,7 +2138,7 @@ termcolor = ">=1.1.0,<2.0.0"
[metadata]
lock-version = "1.1"
python-versions = "^3.9"
-content-hash = "ccce9236755431d057a2fff3fee3a89dfc9de41602ce1b59085e4d036e74e04e"
+content-hash = "36353c45d8dae888096e4bff901be287c14fc1387295075516da8315a659edf9"
[metadata.files]
absl-py = [
@@ -2492,8 +2497,8 @@ flake8-polyfill = [
{file = "flake8_polyfill-1.0.2-py2.py3-none-any.whl", hash = "sha256:12be6a34ee3ab795b19ca73505e7b55826d5f6ad7230d31b18e106400169b9e9"},
]
fsspec = [
- {file = "fsspec-2021.8.1-py3-none-any.whl", hash = "sha256:30f27c059a414d1f434b14b2d0e75c8d9c3dd473ad8daeccb444d9d4069b9f03"},
- {file = "fsspec-2021.8.1.tar.gz", hash = "sha256:af125917788b77782899bbd4484d29e5407f53d2bb04cdfa025fe4931201a555"},
+ {file = "fsspec-2021.9.0-py3-none-any.whl", hash = "sha256:fbddcce8dd2bf7ae348b6930b67a572ba36aa7b1d2a02b122f969e7e9819dd9a"},
+ {file = "fsspec-2021.9.0.tar.gz", hash = "sha256:b5bbe7846c571597e013a917483d561d712ef1e8a13b71629ea8a24c6ffbc3e9"},
]
future = [
{file = "future-0.18.2.tar.gz", hash = "sha256:b1bead90b70cf6ec3f0710ae53a525360fa360d306a86583adc6bf83a4db537d"},
@@ -3499,9 +3504,9 @@ torch = [
{file = "torch-1.9.0-cp39-none-macosx_10_9_x86_64.whl", hash = "sha256:0a9e74b5057463ce4e55d9332a5670993fc9e1299c52e1740e505eda106fb355"},
{file = "torch-1.9.0-cp39-none-macosx_11_0_arm64.whl", hash = "sha256:569ead6ae6bb0e636df0fc8af660ef03260e630dc5f2f4cf3198027e7b6bb481"},
]
-torch-summary = [
- {file = "torch_summary-1.4.5-py3-none-any.whl", hash = "sha256:6127efb631f34ba8cbd28e92e8f7ddc2903ea0cbc44c80f15b7363b8f558cd8c"},
- {file = "torch_summary-1.4.5.tar.gz", hash = "sha256:44eac21777dbbda7b8404d57a43c09d83fd9c93d0c1f0c960b5083ccb24d6d21"},
+torchinfo = [
+ {file = "torchinfo-1.5.3-py3-none-any.whl", hash = "sha256:46ec87260016b70e548e57dcb4a9cc40878c83ab3532fe6a3d6c60437a1ce58f"},
+ {file = "torchinfo-1.5.3.tar.gz", hash = "sha256:49911e5d499f9ff97e270455d1f0d01f205c87cbb7df2bb4c5ba175d88940307"},
]
torchmetrics = [
{file = "torchmetrics-0.4.1-py3-none-any.whl", hash = "sha256:70c83f0fc804a4fe00a9e72dbd2960ff76e39ef62570a19bbdce0c15a1ee0d71"},
diff --git a/pyproject.toml b/pyproject.toml
index 9c972e3..c081a9e 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -20,7 +20,6 @@ matplotlib = "^3.2.1"
tqdm = "^4.46.1"
opencv-python = "^4.3.0"
nltk = "^3.5"
-torch-summary = "^1.4.2"
defusedxml = "^0.6.0"
omegaconf = "^2.1.0"
einops = "^0.3.0"
@@ -57,6 +56,7 @@ flake8-docstrings = "^1.6.0"
darglint = "^1.8.0"
jupyterlab = "^3.0.15"
ipywidgets = "^7.6.3"
+torchinfo = "^1.5.3"
[tool.coverage.report]
fail_under = 50