summaryrefslogtreecommitdiff
path: root/misc/py-torchsummary
diff options
context:
space:
mode:
Diffstat (limited to 'misc/py-torchsummary')
-rw-r--r--misc/py-torchsummary/Makefile21
-rw-r--r--misc/py-torchsummary/distinfo3
-rw-r--r--misc/py-torchsummary/pkg-descr2
3 files changed, 26 insertions, 0 deletions
diff --git a/misc/py-torchsummary/Makefile b/misc/py-torchsummary/Makefile
new file mode 100644
index 000000000000..9ca5d9f08613
--- /dev/null
+++ b/misc/py-torchsummary/Makefile
@@ -0,0 +1,21 @@
+PORTNAME= torchsummary
+DISTVERSION= 1.5.1
+CATEGORIES= misc # machine-learning
+MASTER_SITES= PYPI
+PKGNAMEPREFIX= ${PYTHON_PKGNAMEPREFIX}
+
+MAINTAINER= yuri@FreeBSD.org
+COMMENT= PyTorch: Model summary in PyTorch
+WWW= https://github.com/sksq96/pytorch-summary
+
+LICENSE= MIT
+
+RUN_DEPENDS= ${PYNUMPY} \
+ ${PYTHON_PKGNAMEPREFIX}pytorch>0:misc/py-pytorch@${PY_FLAVOR}
+
+USES= python
+USE_PYTHON= distutils autoplist
+
+NO_ARCH= yes
+
+.include <bsd.port.mk>
diff --git a/misc/py-torchsummary/distinfo b/misc/py-torchsummary/distinfo
new file mode 100644
index 000000000000..1aced87a57d1
--- /dev/null
+++ b/misc/py-torchsummary/distinfo
@@ -0,0 +1,3 @@
+TIMESTAMP = 1758229967
+SHA256 (torchsummary-1.5.1.tar.gz) = 981bf689e22e0cf7f95c746002f20a24ad26aa6b9d861134a14bc6ce92230590
+SIZE (torchsummary-1.5.1.tar.gz) = 3749
diff --git a/misc/py-torchsummary/pkg-descr b/misc/py-torchsummary/pkg-descr
new file mode 100644
index 000000000000..2e23fe04a93e
--- /dev/null
+++ b/misc/py-torchsummary/pkg-descr
@@ -0,0 +1,2 @@
+torchsummary is a Python module that produces model summary in PyTorch similar
+to `model.summary()` in Keras.