summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorAdam J. Stewart <ajstewart426@gmail.com>2021-07-21 14:30:59 -0500
committerGitHub <noreply@github.com>2021-07-21 19:30:59 +0000
commite1b2ab01054719e5d3fcde13e877d64318ce8e2f (patch)
tree056cbb03497af023a0a4ae48b5b6dddb181c6a77
parent147c4fb96d957362918db09adb0d2b4dd6ef548b (diff)
downloadspack-e1b2ab01054719e5d3fcde13e877d64318ce8e2f.tar.gz
spack-e1b2ab01054719e5d3fcde13e877d64318ce8e2f.tar.bz2
spack-e1b2ab01054719e5d3fcde13e877d64318ce8e2f.tar.xz
spack-e1b2ab01054719e5d3fcde13e877d64318ce8e2f.zip
py-pretrainedmodels: add new package (#25020)
-rw-r--r--var/spack/repos/builtin/packages/py-pretrainedmodels/package.py21
1 files changed, 21 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/py-pretrainedmodels/package.py b/var/spack/repos/builtin/packages/py-pretrainedmodels/package.py
new file mode 100644
index 0000000000..64171f11b9
--- /dev/null
+++ b/var/spack/repos/builtin/packages/py-pretrainedmodels/package.py
@@ -0,0 +1,21 @@
+# Copyright 2013-2021 Lawrence Livermore National Security, LLC and other
+# Spack Project Developers. See the top-level COPYRIGHT file for details.
+#
+# SPDX-License-Identifier: (Apache-2.0 OR MIT)
+
+from spack import *
+
+
+class PyPretrainedmodels(PythonPackage):
+ """Pretrained models for Pytorch."""
+
+ homepage = "https://github.com/cadene/pretrained-models.pytorch"
+ pypi = "pretrainedmodels/pretrainedmodels-0.7.4.tar.gz"
+
+ version('0.7.4', sha256='7e77ead4619a3e11ab3c41982c8ad5b86edffe37c87fd2a37ec3c2cc6470b98a')
+
+ depends_on('py-setuptools', type='build')
+ depends_on('py-torch', type=('build', 'run'))
+ depends_on('py-torchvision', type=('build', 'run'))
+ depends_on('py-munch', type=('build', 'run'))
+ depends_on('py-tqdm', type=('build', 'run'))