python312Packages.torchsummary: init at 1.5.1

This commit is contained in:
TomaSajt 2024-05-14 21:51:42 +02:00
parent 9f7053ffc1
commit c1388008b4
No known key found for this signature in database
GPG Key ID: F011163C050122A1
2 changed files with 36 additions and 0 deletions

View File

@ -0,0 +1,34 @@
{
lib,
buildPythonPackage,
fetchPypi,
setuptools,
torch,
}:
buildPythonPackage rec {
pname = "torchsummary";
version = "1.5.1";
pyproject = true;
src = fetchPypi {
inherit pname version;
hash = "sha256-mBv2ieIuDPf5XHRgAvIKJK0mqmudhhE0oUvGzpIjBZA=";
};
build-system = [ setuptools ];
dependencies = [ torch ];
# no tests in pypi tarball
doCheck = false;
pythonImportsCheck = [ "torchsummary" ];
meta = {
description = "Model summary in PyTorch similar to `model.summary()` in Keras";
homepage = "https://github.com/sksq96/pytorch-summary";
license = lib.licenses.mit;
maintainers = with lib.maintainers; [ tomasajt ];
};
}

View File

@ -15306,6 +15306,8 @@ self: super: with self; {
torchsde = callPackage ../development/python-modules/torchsde { };
torchsummary = callPackage ../development/python-modules/torchsummary { };
torchvision = callPackage ../development/python-modules/torchvision { };
torchvision-bin = callPackage ../development/python-modules/torchvision/bin.nix {