forked from fkodom/fft-conv-pytorch
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
29 lines (25 loc) · 848 Bytes
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
from os import name
import setuptools
NAME = "torch_fftconv"
VERSION = '0.1.3'
MAINTAINER = 'Chin-Yun Yu'
EMAIL = '[email protected]'
with open("README.md", "r") as fh:
long_description = fh.read()
setuptools.setup(
name=NAME,
version=VERSION,
author=MAINTAINER,
author_email=EMAIL,
description="Implementation of 1D, 2D, and 3D FFT convolutions in PyTorch. Much faster than direct convolutions for large kernel sizes",
long_description=long_description,
long_description_content_type="text/markdown",
url="https://github.com/yoyololicon/fft-conv-pytorch",
packages=["torch_fftconv"],
install_requires=['torch>=1.7.0'],
classifiers=[
"Programming Language :: Python :: 3",
"License :: OSI Approved :: MIT License",
"Operating System :: OS Independent",
],
)