Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

added option to change default tensor

This commit is contained in:
William Falcon 2019-05-13 22:02:53 -04:00
parent f246ae7fab
commit 8836f4f7a5
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pytorch_lightning/models/trainer.py
View File

@ -32,6 +32,7 @@ class Trainer(TrainerIO):
train_percent_check=1.0, val_percent_check=1.0, test_percent_check=1.0, val_check_interval=0.95, train_percent_check=1.0, val_percent_check=1.0, test_percent_check=1.0, val_check_interval=0.95,
log_save_interval=1, add_log_row_interval=1, log_save_interval=1, add_log_row_interval=1,
lr_scheduler_milestones=None, lr_scheduler_milestones=None,
use_amp=False,
nb_sanity_val_steps=5): nb_sanity_val_steps=5):
# Transfer params # Transfer params
@ -80,7 +81,6 @@ class Trainer(TrainerIO):
print('gpu available: {}, used: {}'.format(torch.cuda.is_available(), self.on_gpu)) print('gpu available: {}, used: {}'.format(torch.cuda.is_available(), self.on_gpu))
# apex test # apex test
use_amp = True
self.use_amp = use_amp and APEX_AVAILABLE self.use_amp = use_amp and APEX_AVAILABLE
if self.use_amp: if self.use_amp:
print('using 16bit precision') print('using 16bit precision')

2
setup.py
View File

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