File size: 650 Bytes
4c1e73e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
from setuptools import setup, find_packages
setup(
name="torchlosses",
version="0.1.0",
description="A collection of advanced PyTorch loss functions (Focal, Dice, Contrastive, Triplet, Cosine, Huber, KLDiv).",
author="Naga Adithya Kaushik (GenAIDevTOProd)",
author_email="adithyakaushikch@gmail.com",
url="https://huggingface.co/GenAIDevTOProd",
packages=find_packages(),
install_requires=["torch>=1.10"],
classifiers=[
"Programming Language :: Python :: 3",
"License :: OSI Approved :: Apache Software License",
"Operating System :: OS Independent",
],
python_requires=">=3.7",
)
|