zhangyiss/spack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

py-pretrainedmodels: add new package (#25020)

Browse Source
This commit is contained in:
Adam J. Stewart 2021-07-21 14:30:59 -05:00 committed by GitHub
parent 147c4fb96d
commit e1b2ab0105
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 21 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
var/spack/repos/builtin/packages/py-pretrainedmodels/package.py Normal file
View File

@ -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'))