wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-09 11:32:07 +08:00
Code Issues Packages Projects Releases Wiki Activity

added option to change default tensor

Browse Source
This commit is contained in:
William Falcon
2019-05-13 20:41:23 -04:00
parent 4693276494
commit fecd6a00cb
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pytorch_lightning/models/trainer.py
+2
View File
@@ -81,6 +81,8 @@ class Trainer(TrainerIO):
# apex test
use_amp = True
self.use_amp = use_amp and APEX_AVAILABLE
if self.use_amp:
print('using 16bit precision')
def __determine_data_use_amount(self, train_percent_check, val_percent_check, test_percent_check, overfit_pct):
"""