added 16 bit training support with --use_amp flag

This commit is contained in:
William Falcon 2019-05-14 05:44:33 -04:00
parent 3b0fd7a6cb
commit 6519c29119
1 changed files with 1 additions and 1 deletions

View File

@ -7,7 +7,7 @@ from setuptools import setup, find_packages
# http://blog.ionelmc.ro/2014/05/25/python-packaging/
setup(
name="pytorch-lightning",
version='0.1.dev1726',
version='0.1.dev18',
description="The Keras for ML researchers using PyTorch",
author="William Falcon",
author_email="waf2107@columbia.edu",