setup.py

from setuptools import find_packages, setup

setup(
    name='transfer_learning',
    packages=find_packages(),
    version='0.0.1',
    description='Transfer learning utility package used in "Effect of large-scale pre-training on full and few-shot transfer learning for natural and medical images"',
    author='Mehdi Cherti, Jenia Jitsev',
    license='MIT',
)